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 []
|