From e3cc20034decf696d291ffd080d860707a131906 Mon Sep 17 00:00:00 2001 From: Macpaul Lin Date: Thu, 8 Jan 2026 22:50:42 +0800 Subject: [PATCH] fix(quant_ops): add _layout_cls and _params aliases to mock QuantizedTensor Signed-off-by: Macpaul Lin --- comfy/quant_ops.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/comfy/quant_ops.py b/comfy/quant_ops.py index 654f5870b..1e0cc0304 100644 --- a/comfy/quant_ops.py +++ b/comfy/quant_ops.py @@ -38,7 +38,9 @@ except ImportError as e: def __init__(self, qdata, layout_type, layout_params): self._qdata = qdata self._layout_type = layout_type + self._layout_cls = layout_type # Alias for compatibility self._layout_params = layout_params + self._params = layout_params # Alias for compatibility self.device = qdata.device self.dtype = qdata.dtype