Update modeling_t5.py
Browse files- 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=
|
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 []
|