From 20675edca60bc29a4aff0640d83c9d61c581395e Mon Sep 17 00:00:00 2001
From: qutcat1997 <54978665+qutcat1997@users.noreply.github.com>
Date: Tue, 11 Jul 2023 23:28:30 +0800
Subject: [PATCH] Update adapter.py

change CFG
---
 pilot/model/adapter.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pilot/model/adapter.py b/pilot/model/adapter.py
index 7721f610f..16179af6b 100644
--- a/pilot/model/adapter.py
+++ b/pilot/model/adapter.py
@@ -114,7 +114,7 @@ class ChatGLMAdapater(BaseLLMAdaper):
     def match(self, model_path: str):
         return "chatglm" in model_path
 
-    def loader(self, model_path: str, from_pretrained_kwargs: dict, device_map=None, num_gpus=cfg.NUM_GPUS):
+    def loader(self, model_path: str, from_pretrained_kwargs: dict, device_map=None, num_gpus=CFG.NUM_GPUS):
         tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
 
         if DEVICE != "cuda":