Update modeling_t5.py

modeling_t5.py

@@ -1,12 +1,3 @@
-from transformers import T5ForConditionalGeneration as t5FCG
-from transformers.models.t5.configuration_t5 import T5Config
-from typing import Optional, Tuple, Union, List, Callable
-
-
-
-
-
-
 class T5ForConditionalGeneration(t5FCG):
     
     def __init__(self, config: T5Config):
@@ -21,7 +12,7 @@ class T5ForConditionalGeneration(t5FCG):
         return text.replace("\\n", "\n").replace("\\t", "\t").replace('%20','  ')
     
     
-    def get_response(self,tokenizer,text, sample=True, top_p=0.9, temperature=0.7,max_length=1024,no_repeat_ngram_size=12,num_beams=1, length_penalty=0.6,):
+    def get_response(self,tokenizer,text, sample=True, top_p=0.9, temperature=0.7,max_length=1024,no_repeat_ngram_size=12,num_beams=1, length_penalty=0.6):
         base_info = "用户：你是谁？\n小元：我是元语智能公司研发的AI智能助手, 在不违反原则的情况下，我可以回答你的任何问题。\n"
         text=base_info+text
         text = self.preprocess(text)
@@ -36,7 +27,7 @@ class T5ForConditionalGeneration(t5FCG):
         return self.postprocess(out_text[0])
     
     
-    def chat(self, tokenizer, query: str, history: List[Tuple[str, str]] = None, sample=True, top_p=0.9, temperature=0.7,max_length=1024):
+    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):
         
         
         history = history or []
@@ -48,7 +39,7 @@ class T5ForConditionalGeneration(t5FCG):
 
         input_text = context + "\n用户：" + query + "\n小元："
         input_text = input_text.strip()
-        response = self.get_response(tokenizer,input_text,sample, top_p, temperature,max_length)
+        response = self.get_response(tokenizer,input_text,sample=sample, top_p=top_p, temperature=temperature,max_length=max_length,no_repeat_ngram_size=no_repeat_ngram_size,num_beams=num_beams, length_penalty=length_penalty)
 
         history.append((query, response))
-        return response,history
+        return response,history