Merge branch 'master' into worksplit-multigpu

2025-04-22 22:40:00 -05:00
parent 6211d2be5a 92cdc692f4
commit 272e8d42c1
27 changed files with 1229 additions and 320 deletions
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -753,6 +753,8 @@ def unet_dtype(device=None, model_params=0, supported_dtypes=[torch.float16, tor
        return torch.float8_e4m3fn
    if args.fp8_e5m2_unet:
        return torch.float8_e5m2
+    if args.fp8_e8m0fnu_unet:
+        return torch.float8_e8m0fnu

    fp8_dtype = None
    if weight_dtype in FLOAT8_TYPES: