From abcfd6bf973fd57c894aa3b3f452ca07ae48b9ed Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Sun, 4 Feb 2024 15:40:02 -0800 Subject: [PATCH] backend --- ldm_patched/modules/model_management.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) 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: