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(backend): add refiner to single-file load_classes

Browse Source 
Fixes single-file refiner loading.
This commit is contained in:
psychedelicious 2024-07-25 19:01:17 +10:00
parent bb876b8d4e
commit 74cef38bcf
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
invokeai/backend/model_manager/load/model_loaders/stable_diffusion.py
View File

@ -98,6 +98,9 @@ class StableDiffusionDiffusersModel(GenericDiffusersLoader):
ModelVariantType.Normal: StableDiffusionXLPipeline, ModelVariantType.Normal: StableDiffusionXLPipeline,
ModelVariantType.Inpaint: StableDiffusionXLInpaintPipeline, ModelVariantType.Inpaint: StableDiffusionXLInpaintPipeline,
}, },
BaseModelType.StableDiffusionXLRefiner: {
ModelVariantType.Normal: StableDiffusionXLPipeline,
},
} }
assert isinstance(config, MainCheckpointConfig) assert isinstance(config, MainCheckpointConfig)
try: try: