From dfa00f17ae94d481ca0f8fd6ce96dbcaa4bbe06e Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Thu, 4 Jul 2024 07:37:17 +0300 Subject: [PATCH] Update. --- reasoning.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/reasoning.py b/reasoning.py index 09e5362..2874adc 100755 --- 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], ) -- 2.39.5