homer7676 commited on
Commit
2536a5e
·
verified ·
1 Parent(s): b693de9

change model path

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -18,11 +18,11 @@ class EndpointHandler:
18
  def initialize(self, context):
19
  """初始化模型和 tokenizer"""
20
  self.tokenizer = AutoTokenizer.from_pretrained(
21
- "THUDM/chatglm3-6b-base",
22
  trust_remote_code=True
23
  )
24
  self.model = AutoModel.from_pretrained(
25
- "THUDM/chatglm3-6b-base",
26
  trust_remote_code=True,
27
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
28
  ).to(self.device)
 
18
  def initialize(self, context):
19
  """初始化模型和 tokenizer"""
20
  self.tokenizer = AutoTokenizer.from_pretrained(
21
+ "homer7676/FrierenChatbotV1",
22
  trust_remote_code=True
23
  )
24
  self.model = AutoModel.from_pretrained(
25
+ "homer7676/FrierenChatbotV1",
26
  trust_remote_code=True,
27
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
28
  ).to(self.device)