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

Add Initial Checks for Inpainting

Browse Source 
The conversion itself is broken. But that's another issue.
This commit is contained in:
blessedcoolant 2023-02-11 20:41:18 +13:00
parent 9232290950
commit 94c31f672f
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
invokeai/backend/invoke_ai_web_server.py
View File

@ -394,13 +394,14 @@ class InvokeAIWebServer:
print("\n") print("\n")
@socketio.on('convertToDiffusers') @socketio.on('convertToDiffusers')
def convert_to_diffusers(model_to_convert: str): def convert_to_diffusers(model_to_convert: dict):
try: try:
if (model_info := self.generate.model_manager.model_info(model_name=model_to_convert)):
if (model_info := self.generate.model_manager.model_info(model_name=model_to_convert['name'])):
if 'weights' in model_info: if 'weights' in model_info:
ckpt_path = Path(model_info['weights']) ckpt_path = Path(model_info['weights'])
original_config_file = Path(model_info['config']) original_config_file = Path(model_info['config'])
model_name = model_to_convert model_name = model_to_convert["name"]
model_description = model_info['description'] model_description = model_info['description']
else: else:
self.socketio.emit("error", {"message": "Model is not a valid checkpoint file"}) self.socketio.emit("error", {"message": "Model is not a valid checkpoint file"})
@ -412,6 +413,13 @@ class InvokeAIWebServer:
if original_config_file and not original_config_file.is_absolute(): if original_config_file and not original_config_file.is_absolute():
original_config_file = Path(Globals.root, original_config_file) original_config_file = Path(Globals.root, original_config_file)
if model_to_convert['is_inpainting']:
original_config_file = Path(
'configs',
'stable-diffusion',
'v1-inpainting-inference.yaml' if model_to_convert['is_inpainting'] else 'v1-inference.yaml'
)
diffusers_path = Path(f'{ckpt_path.parent.absolute()}\\{model_name}_diffusers') diffusers_path = Path(f'{ckpt_path.parent.absolute()}\\{model_name}_diffusers')