kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

change mps dtype to fp32

Browse Source
This commit is contained in:
Qing 2024-01-10 21:48:14 +08:00
parent c3cc7a238e
commit ccfb077a2e
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
iopaint/model/utils.py
View File

@ -1000,7 +1000,10 @@ def get_torch_dtype(device, no_half: bool):
device = str(device)
use_fp16 = not no_half
use_gpu = device == "cuda"
if device in ["cuda", "mps"] and use_fp16:
# https://github.com/huggingface/diffusers/issues/4480
# pipe.enable_attention_slicing and float16 will cause black output on mps
# if device in ["cuda", "mps"] and use_fp16:
if device in ["cuda"] and use_fp16:
return use_gpu, torch.float16
return use_gpu, torch.float32