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…
Reference in a new issue