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

blackify

Browse Source
This commit is contained in:
Lincoln Stein 2023-07-30 19:18:05 -04:00 committed by psychedelicious
parent eeef1e08f8
commit e080fd1e08
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/model_management/models/base.py
View File

@ -292,7 +292,7 @@ class DiffusersModel(ModelBase):
) )
break break
except Exception as e: except Exception as e:
if not str(e).startswith('Error no file'): if not str(e).startswith("Error no file"):
print("====ERR LOAD====") print("====ERR LOAD====")
print(f"{variant}: {e}") print(f"{variant}: {e}")
pass pass

3
invokeai/backend/model_management/models/stable_diffusion.py
View File

@ -22,6 +22,7 @@ import invokeai.backend.util.logging as logger
from invokeai.app.services.config import InvokeAIAppConfig from invokeai.app.services.config import InvokeAIAppConfig
from omegaconf import OmegaConf from omegaconf import OmegaConf
class StableDiffusion1ModelFormat(str, Enum): class StableDiffusion1ModelFormat(str, Enum):
Checkpoint = "checkpoint" Checkpoint = "checkpoint"
Diffusers = "diffusers" Diffusers = "diffusers"
@ -264,7 +265,7 @@ def _convert_ckpt_and_cache(
config_file = app_config.root_path / model_config.config config_file = app_config.root_path / model_config.config
output_path = Path(output_path) output_path = Path(output_path)
variant = model_config.variant variant = model_config.variant
pipeline_class = StableDiffusionInpaintPipeline if variant=='inpaint' else StableDiffusionPipeline pipeline_class = StableDiffusionInpaintPipeline if variant == "inpaint" else StableDiffusionPipeline
# return cached version if it exists # return cached version if it exists
if output_path.exists(): if output_path.exists():