fix comfyui bug

Tencent · Jun 7, 2024 · 38ef213 · 38ef213
1 parent 454c7ba
commit 38ef213
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/comfyui-hydit/hydit/config.py b/comfyui-hydit/hydit/config.py
@@ -20,7 +20,7 @@ def get_args(default_args=None):
     # HunYuan-DiT
     parser_hunyuan.add_argument("--model", type=str, choices=list(HUNYUAN_DIT_CONFIG.keys()), default='DiT-g/2')
     parser_hunyuan.add_argument("--norm", type=str, default="layer", help="Normalization layer type")
-    parser.add_argument("--load-key", type=str, choices=["ema", "module", "distill"], default="ema", help="Load model key for HunYuanDiT checkpoint.")
+    parser_hunyuan.add_argument("--load-key", type=str, choices=["ema", "module", "distill"], default="ema", help="Load model key for HunYuanDiT checkpoint.")
     parser_hunyuan.add_argument('--size-cond', type=int, nargs='+', default=[1024, 1024],
                         help="Size condition used in sampling. 2 values are required for height and width. "
                              "If a single value is provided, the image will be treated to (value, value).")