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; | 	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) => { | export const updateImageSteps = async (token: string = '', steps: number) => { | ||||||
| 	let error = null; | 	let error = null; | ||||||
| 
 | 
 | ||||||
|  |  | ||||||
|  | @ -13,6 +13,7 @@ | ||||||
| 		updateAUTOMATIC1111Url, | 		updateAUTOMATIC1111Url, | ||||||
| 		updateDefaultDiffusionModel, | 		updateDefaultDiffusionModel, | ||||||
| 		updateImageSize, | 		updateImageSize, | ||||||
|  | 		getImageSteps, | ||||||
| 		updateImageSteps | 		updateImageSteps | ||||||
| 	} from '$lib/apis/images'; | 	} from '$lib/apis/images'; | ||||||
| 	import { getBackendConfig } from '$lib/apis'; | 	import { getBackendConfig } from '$lib/apis'; | ||||||
|  | @ -85,6 +86,7 @@ | ||||||
| 
 | 
 | ||||||
| 			if (enableImageGeneration && AUTOMATIC1111_BASE_URL) { | 			if (enableImageGeneration && AUTOMATIC1111_BASE_URL) { | ||||||
| 				imageSize = await getImageSize(localStorage.token); | 				imageSize = await getImageSize(localStorage.token); | ||||||
|  | 				steps = await getImageSteps(localStorage.token); | ||||||
| 				getModels(); | 				getModels(); | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue
	
	 Justin Firth
						Justin Firth