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

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -16,8 +16,8 @@ CONFIG = {
16
 
17
  class EndpointHandler():
18
  def __init__(self,
19
- model_path: str = "",
20
- tokenizer_path: str = ""):
21
 
22
  self.tokenizer = AutoTokenizer.from_pretrained(tokenizer_path)
23
  self.model = AutoModelForSeq2SeqLM.from_pretrained(model_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)
23
  self.model = AutoModelForSeq2SeqLM.from_pretrained(model_path)