Commit a6ef08a4 authored by comfyanonymous's avatar comfyanonymous
Browse files

Even with forced fp16 the cpu device should never use it.

parent 00c0b2c5
...@@ -573,11 +573,15 @@ def should_use_fp16(device=None, model_params=0): ...@@ -573,11 +573,15 @@ def should_use_fp16(device=None, model_params=0):
global xpu_available global xpu_available
global directml_enabled global directml_enabled
if device is not None:
if is_device_cpu(device):
return False
if FORCE_FP16: if FORCE_FP16:
return True return True
if device is not None: #TODO if device is not None: #TODO
if is_device_cpu(device) or is_device_mps(device): if is_device_mps(device):
return False return False
if FORCE_FP32: if FORCE_FP32:
......
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