From 6f4350d65f1e7acffa0fb6b3a93be22192d48eaf Mon Sep 17 00:00:00 2001
From: catboxanon <122327233+catboxanon@users.noreply.github.com>
Date: Thu, 31 Oct 2024 07:09:27 -0400
Subject: [PATCH] Fix typo in .yaml config load (#2228)

---
 backend/loader.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/backend/loader.py b/backend/loader.py
index 0d0b8b4b..72ecc8b7 100644
--- a/backend/loader.py
+++ b/backend/loader.py
@@ -313,7 +313,7 @@ def forge_loader(sd, additional_state_dicts=None):
     has_prediction_type = 'scheduler' in huggingface_components and hasattr(huggingface_components['scheduler'], 'config') and 'prediction_type' in huggingface_components['scheduler'].config
 
     if yaml_config is not None:
-        model_config_params = config.get('model', {}).get('params', {})
+        model_config_params = yaml_config.get('model', {}).get('params', {})
         if "parameterization" in model_config_params:
             if model_config_params["parameterization"] == "v":
                 yaml_config_prediction_type = 'v_prediction'