Update generate.py

generate.py

@@ -105,7 +105,7 @@ def generate(
     do_sample=None,
     early_stopping=None,
     num_beams=None,
-    temperature=None,
+    temperature= 0.7,
     streamer=None,
     top_k=None,
     top_p=None,
@@ -160,12 +160,10 @@ def generate(
     # Set model properties
     self.use_end_thought_token = True
     self.use_start_thought_token = True
-    self.wandb_enabled = True
     self.n_ahead = n_ahead
     self.n_passes = 1
     self.eval_mode = True
     self.first_run = False
-    self.kill_after = 100
     self.rm_initialized = True
     self.original_mode = False