From 82486adf38a242f3f731be53692b9cb5c17e002a Mon Sep 17 00:00:00 2001
From: Great <44393369+yaosheng216@users.noreply.github.com>
Date: Wed, 20 Nov 2024 13:24:28 +0800
Subject: [PATCH] Update model_node.py

---
 scepter/workflow/model_node.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/scepter/workflow/model_node.py b/scepter/workflow/model_node.py
index 99c6555..35e52aa 100644
--- a/scepter/workflow/model_node.py
+++ b/scepter/workflow/model_node.py
@@ -94,13 +94,13 @@ class ModelNode:
             elif source == 'Local':
                 cfg_new = copy.deepcopy(cfg)
                 cfg_new.MODEL = cfg_new.MODEL_LOCAL
-                if hasattr(cfg_new, 'EFINER_MODEL_LOCAL'):
+                if hasattr(cfg_new, 'REFINER_MODEL_LOCAL'):
                     cfg_new.REFINER_MODEL = cfg_new.REFINER_MODEL_LOCAL
                 return cfg_new
             elif source == 'HuggingFace':
                 cfg_new = copy.deepcopy(cfg)
                 cfg_new.MODEL = cfg_new.MODEL_HF
-                if hasattr(cfg_new, 'EFINER_MODEL_HF'):
+                if hasattr(cfg_new, 'REFINER_MODEL_HF'):
                     cfg_new.REFINER_MODEL = cfg_new.REFINER_MODEL_HF
                 return cfg_new
             else: