LiuZichen commited on
Commit
d647b76
·
verified ·
1 Parent(s): e009812

Update MagicQuill/comfy/cli_args.py

Browse files
Files changed (1) hide show
  1. MagicQuill/comfy/cli_args.py +1 -1
MagicQuill/comfy/cli_args.py CHANGED
@@ -59,7 +59,7 @@ fp_group.add_argument("--force-fp16", action="store_true", help="Force fp16.")
59
  fpunet_group = parser.add_mutually_exclusive_group()
60
  fpunet_group.add_argument("--bf16-unet", action="store_true", help="Run the UNET in bf16. This should only be used for testing stuff.")
61
  fpunet_group.add_argument("--fp16-unet", action="store_true", help="Store unet weights in fp16.")
62
- fpunet_group.add_argument("--fp8_e4m3fn-unet", action="store_true", help="Store unet weights in fp8_e4m3fn.")
63
  fpunet_group.add_argument("--fp8_e5m2-unet", action="store_true", help="Store unet weights in fp8_e5m2.")
64
 
65
  fpvae_group = parser.add_mutually_exclusive_group()
 
59
  fpunet_group = parser.add_mutually_exclusive_group()
60
  fpunet_group.add_argument("--bf16-unet", action="store_true", help="Run the UNET in bf16. This should only be used for testing stuff.")
61
  fpunet_group.add_argument("--fp16-unet", action="store_true", help="Store unet weights in fp16.")
62
+ fpunet_group.add_argument("--fp8_e4m3fn-unet", type=bool, default=True, help="Store unet weights in fp8_e4m3fn.")
63
  fpunet_group.add_argument("--fp8_e5m2-unet", action="store_true", help="Store unet weights in fp8_e5m2.")
64
 
65
  fpvae_group = parser.add_mutually_exclusive_group()