diff --git a/invokeai/backend/model_management/convert_ckpt_to_diffusers.py b/invokeai/backend/model_management/convert_ckpt_to_diffusers.py index d3dee08b75..0e8ac99472 100644 --- a/invokeai/backend/model_management/convert_ckpt_to_diffusers.py +++ b/invokeai/backend/model_management/convert_ckpt_to_diffusers.py @@ -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 diff --git a/invokeai/backend/model_management/model_cache.py b/invokeai/backend/model_management/model_cache.py index dd7e04cc31..4e1cc2a16b 100644 --- a/invokeai/backend/model_management/model_cache.py +++ b/invokeai/backend/model_management/model_cache.py @@ -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,