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: Update missing name types to new names

Browse Source
This commit is contained in:
blessedcoolant 2023-06-17 22:04:28 +12:00
parent 76dd749b1e
commit c8dfa49d86
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -107,7 +107,7 @@ class StableDiffusion1Model(DiffusersModel):
) -> str: ) -> str:
assert model_path == config.path assert model_path == config.path
if isinstance(config, cls.CheckpointModelConfig): if isinstance(config, cls.StableDiffusion1CheckpointModelConfig):
return _convert_ckpt_and_cache( return _convert_ckpt_and_cache(
version=BaseModelType.StableDiffusion1, version=BaseModelType.StableDiffusion1,
model_config=config, model_config=config,
@ -220,7 +220,7 @@ class StableDiffusion2Model(DiffusersModel):
) -> str: ) -> str:
assert model_path == config.path assert model_path == config.path
if isinstance(config, cls.CheckpointModelConfig): if isinstance(config, cls.StableDiffusion2CheckpointModelConfig):
return _convert_ckpt_and_cache( return _convert_ckpt_and_cache(
version=BaseModelType.StableDiffusion2, version=BaseModelType.StableDiffusion2,
model_config=config, model_config=config,