Linear-Matrix-Probability commited on
Commit
6bfc699
·
verified ·
1 Parent(s): abfdef5

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -16,7 +16,7 @@ CONFIG = {
16
 
17
  class EndpointHandler():
18
  def __init__(self,
19
- model_path: str = "zjunlp/chatcell-small,
20
  tokenizer_path: str = "zjunlp/chatcell-small"):
21
 
22
  self.tokenizer = AutoTokenizer.from_pretrained(tokenizer_path)
 
16
 
17
  class EndpointHandler():
18
  def __init__(self,
19
+ model_path: str = "zjunlp/chatcell-small",
20
  tokenizer_path: str = "zjunlp/chatcell-small"):
21
 
22
  self.tokenizer = AutoTokenizer.from_pretrained(tokenizer_path)