X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=quiz_machine.py;h=f0fb4082beb73913460bd4f72d4448e22ed15a1c;hb=00f7b3d445af8bb57376faabbf74eadc145faf1f;hp=9f4fe9606cfa4a657b81c3197fd36160a2def394;hpb=14c70d068471a163ecd389e0c1667e561ea056f9;p=culture.git

diff --git a/quiz_machine.py b/quiz_machine.py
index 9f4fe96..f0fb408 100755
--- a/quiz_machine.py
+++ b/quiz_machine.py
@@ -27,8 +27,8 @@ def one_batch_masked_inplace_autoregression(
     input,
     ar_mask,
     seq_logproba,
-    temperature=1.0,
-    deterministic_synthesis=False,
+    temperature,
+    deterministic_synthesis,
 ):
     to_generate = (ar_mask.sum(0) > 0).nonzero()