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

if sdxl-vae-fp16-fix model is available then bake it in when converting ckpts

Browse Source
This commit is contained in:
Lincoln Stein 2023-09-05 08:18:51 -04:00 committed by Kent Keirsey
parent 604f0083f2
commit 4e83644433
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
invokeai/backend/model_management/models/sdxl.py
View File

@ -119,11 +119,22 @@ class StableDiffusionXLModel(DiffusersModel):
if isinstance(config, cls.CheckpointConfig):
from invokeai.backend.model_management.models.stable_diffusion import _convert_ckpt_and_cache
# Hack in VAE-fp16 fix - If model sdxl-vae-fp16-fix is installed,
# then we bake it into the converted model.
from invokeai.app.services.config import InvokeAIAppConfig
kwargs = dict()
app_config = InvokeAIAppConfig.get_config()
vae_path = app_config.models_path / 'sdxl/vae/sdxl-vae-fp16-fix'
if vae_path.exists():
kwargs['vae_path'] = vae_path
return _convert_ckpt_and_cache(
version=base_model,
model_config=config,
output_path=output_path,
use_safetensors=False, # corrupts sdxl models for some reason
**kwargs,
)
else:
return model_path