Only auto enable bf16 VAE on nvidia GPUs that actually support it.

1 year ago · f9e55d8463
parent 2395ae740a
commit f9e55d8463
1 changed files with 1 additions and 1 deletions
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@ -175,7 +175,7 @@ try:
        if int(torch_version[0]) >= 2:
            if ENABLE_PYTORCH_ATTENTION == False and args.use_split_cross_attention == False and args.use_quad_cross_attention == False:
                ENABLE_PYTORCH_ATTENTION = True
-            if torch.cuda.is_bf16_supported():
+            if torch.cuda.is_bf16_supported() and torch.cuda.get_device_properties(torch.cuda.current_device()).major >= 8:
                VAE_DTYPE = torch.bfloat16
    if is_intel_xpu():
        if args.use_split_cross_attention == False and args.use_quad_cross_attention == False: