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
a6ef08a4
"docs/vscode:/vscode.git/clone" did not exist on "02ef825be208badb6bb7bf0641e7035406690b18"
Commit
a6ef08a4
authored
Aug 23, 2023
by
comfyanonymous
Browse files
Even with forced fp16 the cpu device should never use it.
parent
00c0b2c5
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
1 deletion
+5
-1
comfy/model_management.py
comfy/model_management.py
+5
-1
No files found.
comfy/model_management.py
View file @
a6ef08a4
...
@@ -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
:
...
...
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