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

Revert "fix(mm): provide ckpt config as stream to diffusers"

Browse Source 
This reverts commit 9d045964d6d7157fa8162001e1f6badc6fe2ba48.
This commit is contained in:
Lincoln Stein 2024-03-18 22:44:04 -04:00 committed by psychedelicious
parent 9d045964d6
commit f8df293d2c
1 changed files with 13 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -79,18 +79,17 @@ 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, prediction_type=prediction_type,
prediction_type=prediction_type, image_size=image_size,
image_size=image_size, upcast_attention=upcast_attention,
upcast_attention=upcast_attention, load_safety_checker=False,
load_safety_checker=False, )
)
return output_path return output_path