Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix tests

Browse Source
This commit is contained in:
Jennifer Player
2024-03-06 14:26:53 -05:00
committed by Kent Keirsey
parent 4af5a09a68
commit 347f1fd0b7
4 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
invokeai/app/api/routers/model_manager.py
View File

@ -79,7 +79,6 @@ example_model_input = {
"description": "Model description", "description": "Model description",
"vae": None, "vae": None,
"variant": "normal", "variant": "normal",
"image": "blob",
} }
############################################################################## ##############################################################################

6
invokeai/frontend/web/src/services/api/endpoints/models.ts
View File

@ -24,8 +24,8 @@ export type UpdateModelArg = {
}; };
export type UpdateModelImageArg = { export type UpdateModelImageArg = {
key: paths['/api/v2/models/i/{key}/image']['patch']['parameters']['path']['key']; key: string;
image: paths['/api/v2/models/i/{key}/image']['patch']['requestBody']['content']['multipart/form-data']; image: Blob;
}; };
type UpdateModelResponse = paths['/api/v2/models/i/{key}']['patch']['responses']['200']['content']['application/json']; type UpdateModelResponse = paths['/api/v2/models/i/{key}']['patch']['responses']['200']['content']['application/json'];
@ -155,7 +155,7 @@ export const modelsApi = api.injectEndpoints({
updateModelImage: build.mutation<UpdateModelImageResponse, UpdateModelImageArg>({ updateModelImage: build.mutation<UpdateModelImageResponse, UpdateModelImageArg>({
query: ({ key, image }) => { query: ({ key, image }) => {
const formData = new FormData(); const formData = new FormData();
formData.append('image', image.image); formData.append('image', image);
return { return {
url: buildModelsUrl(`i/${key}/image`), url: buildModelsUrl(`i/${key}/image`),
method: 'PATCH', method: 'PATCH',

7
invokeai/frontend/web/src/services/api/schema.ts
View File

File diff suppressed because one or more lines are too long

1
tests/conftest.py
View File

@ -44,6 +44,7 @@ def mock_services() -> InvocationServices:
images=ImageService(), images=ImageService(),
invocation_cache=MemoryInvocationCache(max_cache_size=0), invocation_cache=MemoryInvocationCache(max_cache_size=0),
logger=logging, # type: ignore logger=logging, # type: ignore
model_images=None, # type: ignore
model_manager=None, # type: ignore model_manager=None, # type: ignore
download_queue=None, # type: ignore download_queue=None, # type: ignore
names=None, # type: ignore names=None, # type: ignore