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

Fixes for checkpoint models

Browse Source
This commit is contained in:
Sergey Borisov 2023-05-30 19:12:47 +03:00
parent 420a76ecdd
commit 69ccd3a0b5
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/model_management/convert_ckpt_to_diffusers.py
View File

@ -1320,7 +1320,7 @@ def load_pipeline_from_original_stable_diffusion_ckpt(
)
safety_checker = StableDiffusionSafetyChecker.from_pretrained(
"CompVis/stable-diffusion-safety-checker",
cache_dir=config.cache_dir,
cache_dir=cache_dir,
)
feature_extractor = AutoFeatureExtractor.from_pretrained(
"CompVis/stable-diffusion-safety-checker", cache_dir=cache_dir

2
invokeai/backend/model_management/model_cache.py
View File

@ -393,7 +393,7 @@ class VaeModelInfo(ModelInfoBase):
if child_type is not None:
raise Exception("There is no child models in vae model")
model = self.vae_type.from_pretrained(
model = self.vae_class.from_pretrained(
self.repo_id_or_path,
cache_dir=get_invokeai_config().cache_dir,
torch_dtype=torch_dtype,