From 44d9e1ed1f3e43e754e6ef900d418bf049234e8f Mon Sep 17 00:00:00 2001
From: minsoolee <joylms99@naver.com>
Date: Thu, 26 Jun 2025 15:53:14 +0900
Subject: [PATCH] Fix: Use config variable for 'more_smooth' parameter to avoid
 hard-coding

---
 demo_zero_shot_edit.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/demo_zero_shot_edit.ipynb b/demo_zero_shot_edit.ipynb
index a186be3..f30c673 100644
--- a/demo_zero_shot_edit.ipynb
+++ b/demo_zero_shot_edit.ipynb
@@ -282,7 +282,7 @@
     "    with torch.autocast('cuda', enabled=True, dtype=torch.float16, cache_enabled=True):    # using bfloat16 can be faster\n",
     "        recon_B3HW = autoregressive_infer_cfg_with_mask(\n",
     "            var,\n",
-    "            B=B, label_B=label_B, cfg=3, top_k=900, top_p=0.95, g_seed=0, more_smooth=True,\n",
+    "            B=B, label_B=label_B, cfg=3, top_k=900, top_p=0.95, g_seed=0, more_smooth=more_smooth,\n",
     "            input_img_tokens=input_img_tokens, edit_mask=edit_mask\n",
     "        )\n",
     "\n",