Update.

diff --git a/reasoning.py b/reasoning.py
index 09e5362..2874adc 100755 (executable)
--- a/reasoning.py
+++ b/reasoning.py
@@ -467,8 +467,8 @@ if __name__ == "__main__":
     delay = time.perf_counter() - start_time
     print(f"{prompts.size(0)/delay:02f} seq/s")
 
-    # predicted_prompts = torch.rand(prompts.size(0)) < 0.5
-    # predicted_answers = torch.logical_not(predicted_prompts)
+    predicted_prompts = torch.rand(prompts.size(0)) < 0.5
+    predicted_answers = torch.logical_not(predicted_prompts)
 
     reasoning.save_quizzes(
         "/tmp",
@@ -476,5 +476,6 @@ if __name__ == "__main__":
         prompts[:64],
         answers[:64],
         # You can add a bool to put a frame around the predicted parts
-        # predicted_prompts, predicted_answers
+        predicted_prompts[:64],
+        predicted_answers[:64],
     )