diff --git a/ldm_patched/modules/model_management.py b/ldm_patched/modules/model_management.py index c9d9f52f..7859d39d 100644 --- a/ldm_patched/modules/model_management.py +++ b/ldm_patched/modules/model_management.py @@ -560,10 +560,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 + return torch.float16 + def intermediate_device(): if args.always_gpu: