lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-09 09:29:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Only use .yaml config prediction if actually set (#2272)

Browse Source
This commit is contained in:
catboxanon
2024-11-06 23:44:12 -05:00
committed by GitHub
parent 75120d02f3
commit f4afbaff45
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend/loader.py
View File

@@ -321,7 +321,7 @@ def forge_loader(sd, additional_state_dicts=None):
yaml_config_prediction_type = 'v_prediction'
if has_prediction_type:
if yaml_config_prediction_type is not None:
if yaml_config_prediction_type:
huggingface_components['scheduler'].config.prediction_type = yaml_config_prediction_type
else:
huggingface_components['scheduler'].config.prediction_type = prediction_types.get(estimated_config.model_type.name, huggingface_components['scheduler'].config.prediction_type)