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 "unknown model type" error when rebasing a model with API

Browse Source 
- Add command-line model probing script for dev use
- Minor documentation tweak
This commit is contained in:
Lincoln Stein
2023-07-25 08:36:57 -04:00
parent 8c449c4756
commit 2aefa921fe
4 changed files with 27 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/app/api/routers/models.py
View File

@ -87,7 +87,7 @@ async def update_model(
new_name = info.model_name,
new_base = info.base_model,
)
logger.info(f'Successfully renamed {base_model}/{model_name}=>{info.base_model}/{info.model_name}')
logger.info(f'Successfully renamed {base_model.value}/{model_name}=>{info.base_model}/{info.model_name}')
# update information to support an update of attributes
model_name = info.model_name
base_model = info.base_model
@ -98,7 +98,7 @@ async def update_model(
)
if new_info.get('path') != previous_info.get('path'): # model manager moved model path during rename - don't overwrite it
info.path = new_info.get('path')
ApiDependencies.invoker.services.model_manager.update_model(
model_name=model_name,
base_model=base_model,