ClueAI commited on
Commit
b1373ff
1 Parent(s): 6b9e4d0

Update modeling_t5.py

Browse files
Files changed (1) hide show
  1. modeling_t5.py +1 -1
modeling_t5.py CHANGED
@@ -36,7 +36,7 @@ class T5ForConditionalGeneration(t5FCG):
36
  return self.postprocess(out_text[0])
37
 
38
 
39
- def chat(self, tokenizer, query: str, history: List[Tuple[str, str]] = None, sample=True, top_p=0.9, temperature=0.7,max_length=1024,no_repeat_ngram_size=12,num_beams=1, length_penalty=0.6):
40
 
41
 
42
  history = history or []
 
36
  return self.postprocess(out_text[0])
37
 
38
 
39
+ def chat(self, tokenizer, query: str, history: List[Tuple[str, str]] = None, sample=True, top_p=0.9, temperature=0.7,max_length=2048,no_repeat_ngram_size=12,num_beams=1, length_penalty=0.6):
40
 
41
 
42
  history = history or []