This commit is contained in:
lllyasviel
2024-02-04 15:40:02 -08:00
parent cf0bb69a27
commit abcfd6bf97

View File

@@ -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: