Always use fp16 for the text encoders.

4b023906 · comfyanonymous · d0e2354c · 4b023906
Commit 4b023906 authored Feb 02, 2024 by comfyanonymous
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 4 deletions

comfy/model_management.py comfy/model_management.py +2 -4

No files found.
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -546,10 +546,8 @@ def text_encoder_dtype(device=None):
    if is_device_cpu(device):
        return torch.float16

-    if should_use_fp16(device, prioritize_performance=False):
    return torch.float16
-    else:
-        return torch.float32
+

 def intermediate_device():
    if args.gpu_only: