diff --git a/comfy/ops.py b/comfy/ops.py index 079309972..279f6b1a7 100644 --- a/comfy/ops.py +++ b/comfy/ops.py @@ -530,7 +530,7 @@ if CUBLAS_IS_AVAILABLE: # ============================================================================== # Mixed Precision Operations # ============================================================================== -from .quant_ops import QuantizedTensor, TensorCoreFP8Layout +from .quant_ops import QuantizedTensor QUANT_FORMAT_MIXINS = { "float8_e4m3fn": {