forked from open-webui/open-webui
		
	Restore image steps setting when Settings modal is presented
This commit is contained in:
		
							parent
							
								
									3fa79e59bf
								
							
						
					
					
						commit
						af1675813f
					
				
					 2 changed files with 34 additions and 0 deletions
				
			
		|  | @ -198,6 +198,38 @@ export const updateImageSize = async (token: string = '', size: string) => { | |||
| 	return res.IMAGE_SIZE; | ||||
| }; | ||||
| 
 | ||||
| export const getImageSteps = async (token: string = '') => { | ||||
| 	let error = null; | ||||
| 
 | ||||
| 	const res = await fetch(`${IMAGES_API_BASE_URL}/steps`, { | ||||
| 		method: 'GET', | ||||
| 		headers: { | ||||
| 			Accept: 'application/json', | ||||
| 			'Content-Type': 'application/json', | ||||
| 			...(token && { authorization: `Bearer ${token}` }) | ||||
| 		} | ||||
| 	}) | ||||
| 		.then(async (res) => { | ||||
| 			if (!res.ok) throw await res.json(); | ||||
| 			return res.json(); | ||||
| 		}) | ||||
| 		.catch((err) => { | ||||
| 			console.log(err); | ||||
| 			if ('detail' in err) { | ||||
| 				error = err.detail; | ||||
| 			} else { | ||||
| 				error = 'Server connection failed'; | ||||
| 			} | ||||
| 			return null; | ||||
| 		}); | ||||
| 
 | ||||
| 	if (error) { | ||||
| 		throw error; | ||||
| 	} | ||||
| 
 | ||||
| 	return res.IMAGE_STEPS; | ||||
| }; | ||||
| 
 | ||||
| export const updateImageSteps = async (token: string = '', steps: number) => { | ||||
| 	let error = null; | ||||
| 
 | ||||
|  |  | |||
|  | @ -13,6 +13,7 @@ | |||
| 		updateAUTOMATIC1111Url, | ||||
| 		updateDefaultDiffusionModel, | ||||
| 		updateImageSize, | ||||
| 		getImageSteps, | ||||
| 		updateImageSteps | ||||
| 	} from '$lib/apis/images'; | ||||
| 	import { getBackendConfig } from '$lib/apis'; | ||||
|  | @ -85,6 +86,7 @@ | |||
| 
 | ||||
| 			if (enableImageGeneration && AUTOMATIC1111_BASE_URL) { | ||||
| 				imageSize = await getImageSize(localStorage.token); | ||||
| 				steps = await getImageSteps(localStorage.token); | ||||
| 				getModels(); | ||||
| 			} | ||||
| 		} | ||||
|  |  | |||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue
	
	 Justin Firth
						Justin Firth