298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-19 19:03:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Always use fp16 for the text encoders.

Browse Source
This commit is contained in:
comfyanonymous 2024-02-02 10:02:49 -05:00
parent d0e2354c28
commit 4b0239066d
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/model_management.py
View File

@ -546,10 +546,8 @@ def text_encoder_dtype(device=None):
if is_device_cpu(device): if is_device_cpu(device):
return torch.float16 return torch.float16
if should_use_fp16(device, prioritize_performance=False): return torch.float16
return torch.float16
else:
return torch.float32
def intermediate_device(): def intermediate_device():
if args.gpu_only: if args.gpu_only: