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(mm): provide ckpt config as stream to diffusers

Browse Source 
Fixes converting ckpt main models
This commit is contained in:
psychedelicious 2024-03-18 22:39:15 +11:00
parent 9fa9ebe386
commit 9d045964d6
1 changed files with 14 additions and 13 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

@ -79,11 +79,12 @@ class StableDiffusionDiffusersModel(GenericDiffusersLoader):
) )
self._logger.info(f"Converting {model_path} to diffusers format") self._logger.info(f"Converting {model_path} to diffusers format")
with open(self._app_config.root_path / config_file, "r") as config_stream:
convert_ckpt_to_diffusers( convert_ckpt_to_diffusers(
model_path, model_path,
output_path, output_path,
model_type=self.model_base_to_model_type[base], model_type=self.model_base_to_model_type[base],
original_config_file=self._app_config.root_path / config_file, original_config_file=config_stream,
extract_ema=True, extract_ema=True,
from_safetensors=model_path.suffix == ".safetensors", from_safetensors=model_path.suffix == ".safetensors",
precision=self._torch_dtype, precision=self._torch_dtype,