update

simulator.py

@@ -15,22 +15,13 @@ class Simulator:
             device_map="auto"
         )
         self.model.eval()
-
         self.generation_kwargs = dict(
-            do_sample=False,
+            do_sample=True,
             temperature=0.7,
             max_length=500,
             max_new_tokens=10
         )
 
-        generation_kwargs = dict(
-
-            max_length=500,
-            max_new_tokens=200
-        )
-
-        print(1)
-
     def generate_query(self, history):
         """ user simulator
         :param history: