From f4afbaff456e6b04ebe66a7562e1c710b059b8ca Mon Sep 17 00:00:00 2001 From: catboxanon <122327233+catboxanon@users.noreply.github.com> Date: Wed, 6 Nov 2024 23:44:12 -0500 Subject: [PATCH] Only use .yaml config prediction if actually set (#2272) --- backend/loader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/loader.py b/backend/loader.py index 5eff980b..ad2834a0 100644 --- a/backend/loader.py +++ b/backend/loader.py @@ -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)