Skip to content

GitLab

Commit 0d7b0a4d authored by comfyanonymous's avatar comfyanonymous
Browse files

Small cleanups.

parent 92254659
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
comfy/model_management.py
View file @ 0d7b0a4d
...@@ -589,7 +589,10 @@ def should_use_fp16(device=None, model_params=0): ...@@ -589,7 +589,10 @@ def should_use_fp16(device=None, model_params=0):
if cpu_mode() or mps_mode(): if cpu_mode() or mps_mode():
return False #TODO ? return False #TODO ?
if torch.cuda.is_bf16_supported() or xpu_available: if xpu_available:
return True
if torch.cuda.is_bf16_supported():
return True return True
props = torch.cuda.get_device_properties("cuda") props = torch.cuda.get_device_properties("cuda")
......
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