Skip to content

GitLab

Commit 4b023906 authored by comfyanonymous's avatar comfyanonymous
Browse files

Always use fp16 for the text encoders.

parent d0e2354c
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 4 deletions
comfy/model_management.py
View file @ 4b023906
...@@ -546,10 +546,8 @@ def text_encoder_dtype(device=None): ...@@ -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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment