From 6070c4a1e54d01a2f95e707156069ab36c2205b4 Mon Sep 17 00:00:00 2001
From: lvmin <lyuminzhang@outlook.com>
Date: Wed, 9 Aug 2023 13:41:55 -0700
Subject: [PATCH] i

---
 entry.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/entry.py b/entry.py
index 758fef73..471cd20d 100644
--- a/entry.py
+++ b/entry.py
@@ -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(