wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-25 19:13:31 +08:00
Code Issues Actions 27 Packages Projects Releases Wiki Activity

Guards

Browse Source
This commit is contained in:
kijai 2026-03-12 19:11:21 +02:00
parent d59731cf26
commit 1f6691077d
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
comfy/ops.py
View File

@ -975,6 +975,7 @@ def pick_operations(weight_dtype, compute_dtype, load_device=None, disable_fast_
if not fp8_compute: if not fp8_compute:
disabled.add("float8_e4m3fn") disabled.add("float8_e4m3fn")
disabled.add("float8_e5m2") disabled.add("float8_e5m2")
disabled.add("mxfp8")
return mixed_precision_ops(model_config.quant_config, compute_dtype, disabled=disabled) return mixed_precision_ops(model_config.quant_config, compute_dtype, disabled=disabled)
if ( if (

6
comfy/quant_ops.py
View File

@ -38,6 +38,9 @@ except ImportError as e:
class _CKNvfp4Layout: class _CKNvfp4Layout:
pass pass
class _CKMxfp8Layout:
pass
def register_layout_class(name, cls): def register_layout_class(name, cls):
pass pass
@ -138,7 +141,8 @@ register_layout_class("TensorCoreFP8Layout", TensorCoreFP8Layout)
register_layout_class("TensorCoreFP8E4M3Layout", TensorCoreFP8E4M3Layout) register_layout_class("TensorCoreFP8E4M3Layout", TensorCoreFP8E4M3Layout)
register_layout_class("TensorCoreFP8E5M2Layout", TensorCoreFP8E5M2Layout) register_layout_class("TensorCoreFP8E5M2Layout", TensorCoreFP8E5M2Layout)
register_layout_class("TensorCoreNVFP4Layout", TensorCoreNVFP4Layout) register_layout_class("TensorCoreNVFP4Layout", TensorCoreNVFP4Layout)
register_layout_class("TensorCoreMXFP8Layout", TensorCoreMXFP8Layout) if _CK_AVAILABLE:
register_layout_class("TensorCoreMXFP8Layout", TensorCoreMXFP8Layout)
QUANT_ALGOS = { QUANT_ALGOS = {
"float8_e4m3fn": { "float8_e4m3fn": {