i

2026-08-16 13:13:16 +02:00 · 2023-08-09 13:41:55 -07:00
parent 5bf4a08b5e
commit 6070c4a1e5
1 changed files with 1 additions and 1 deletions
@@ -106,7 +106,7 @@ value_dict = {
 batch, batch_uc = get_batch(
    get_unique_embedder_keys_from_conditioner(model.conditioner),
    value_dict,
-    1,
+    [1],
 )

 c, uc = model.conditioner.get_unconditional_conditioning(