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)