Skip to content

Commit a60620d

Browse files
Fix slow performance on 10 series Nvidia GPUs.
1 parent 015f73d commit a60620d

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

comfy/model_management.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -668,6 +668,7 @@ def unet_manual_cast(weight_dtype, inference_device, supported_dtypes=[torch.flo
668668
if bf16_supported and weight_dtype == torch.bfloat16:
669669
return None
670670

671+
fp16_supported = should_use_fp16(inference_device, prioritize_performance=True)
671672
for dt in supported_dtypes:
672673
if dt == torch.float16 and fp16_supported:
673674
return torch.float16

0 commit comments

Comments
 (0)