Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
ComfyUI
Commits
88f30040
Commit
88f30040
authored
Feb 19, 2024
by
comfyanonymous
Browse files
Enable fp16 by default on mps.
parent
e93cdd0a
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
4 deletions
+7
-4
comfy/model_management.py
comfy/model_management.py
+7
-4
No files found.
comfy/model_management.py
View file @
88f30040
...
...
@@ -720,9 +720,9 @@ def should_use_fp16(device=None, model_params=0, prioritize_performance=True, ma
if
FORCE_FP16
:
return
True
if
device
is
not
None
:
#TODO
if
device
is
not
None
:
if
is_device_mps
(
device
):
return
Fals
e
return
Tru
e
if
FORCE_FP32
:
return
False
...
...
@@ -730,8 +730,11 @@ def should_use_fp16(device=None, model_params=0, prioritize_performance=True, ma
if
directml_enabled
:
return
False
if
cpu_mode
()
or
mps_mode
():
return
False
#TODO ?
if
mps_mode
():
return
True
if
cpu_mode
():
return
False
if
is_intel_xpu
():
return
True
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment