From 6e1a7908b444af40165986f383933a0d0e52ff6d Mon Sep 17 00:00:00 2001 From: catboxanon <122327233+catboxanon@users.noreply.github.com> Date: Wed, 6 Nov 2024 23:49:18 -0500 Subject: [PATCH] Fallback to estimated prediction if .yaml prediction not suitable (#2273) --- backend/loader.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/backend/loader.py b/backend/loader.py index ad2834a0..d5b6ca90 100644 --- a/backend/loader.py +++ b/backend/loader.py @@ -319,6 +319,9 @@ def forge_loader(sd, additional_state_dicts=None): ) if yaml_config_prediction_type == 'v' or yaml_config_prediction_type.endswith(".VScaling"): yaml_config_prediction_type = 'v_prediction' + else: + # Use estimated prediction config if no suitable prediction type found + yaml_config_prediction_type = '' if has_prediction_type: if yaml_config_prediction_type: