Spaces:
Running
on
L4
Running
on
L4
Update MagicQuill/comfy/cli_args.py
Browse files
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",
|
| 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()
|