Hermes Bot commited on
Commit ·
bec4723
1
Parent(s): 6fdb441
Make should_use_bf16 stub accept manual_cast argument
Browse files
comfy_integration/__pycache__/setup.cpython-311.pyc
CHANGED
|
Binary files a/comfy_integration/__pycache__/setup.cpython-311.pyc and b/comfy_integration/__pycache__/setup.cpython-311.pyc differ
|
|
|
comfy_integration/setup.py
CHANGED
|
@@ -172,7 +172,7 @@ def initialize_comfyui():
|
|
| 172 |
import importlib
|
| 173 |
importlib.reload(model_mgmt)
|
| 174 |
# Override should_use_bf16 to avoid any CUDA device property queries on CPU‑only systems.
|
| 175 |
-
def _safe_should_use_bf16(device, model_params=None):
|
| 176 |
"""Return False unconditionally – no BF16 support on CPU‑only environment."""
|
| 177 |
return False
|
| 178 |
model_mgmt.should_use_bf16 = _safe_should_use_bf16
|
|
|
|
| 172 |
import importlib
|
| 173 |
importlib.reload(model_mgmt)
|
| 174 |
# Override should_use_bf16 to avoid any CUDA device property queries on CPU‑only systems.
|
| 175 |
+
def _safe_should_use_bf16(device, model_params=None, manual_cast=False):
|
| 176 |
"""Return False unconditionally – no BF16 support on CPU‑only environment."""
|
| 177 |
return False
|
| 178 |
model_mgmt.should_use_bf16 = _safe_should_use_bf16
|