Merge pull request #63 from yaosheng216/patch-5

Update model_node.py
2024-11-20 13:26:33 +08:00
parent a683061c6f 82486adf38
commit d48b2f110f
1 changed files with 2 additions and 2 deletions
@@ -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: