From: François Fleuret <francois@fleuret.org>
Date: Mon, 22 Jul 2024 04:38:40 +0000 (+0200)
Subject: Update.
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=197703b2e1d388a380634522dd2c52182d585028;p=culture.git

Update.
---

diff --git a/quiz_machine.py b/quiz_machine.py
index 5f14528..c73b6d0 100755
--- a/quiz_machine.py
+++ b/quiz_machine.py
@@ -73,7 +73,8 @@ def one_batch_masked_inplace_autoregression(
 
         logits = output[:, s]
 
-        logits = logit_transformer(s, logits).log_softmax(dim=-1)
+        if logit_transformer is not None:
+            logits = logit_transformer(s, logits).log_softmax(dim=-1)
 
         if deterministic_synthesis:
             t_next = logits.argmax(-1)