Add __init__.py for group wise lut quantization package

szyszyzys · liangel-02 · commit 01b81bc362bc · 2025-08-25T11:15:08.000-07:00
Differential Revision: D79119958 Pull Request resolved: #2702
diff --git a/torchao/prototype/quantization/codebook_groupwise/__init__.py b/torchao/prototype/quantization/codebook_groupwise/__init__.py
@@ -0,0 +1,4 @@
+from .api import GroupwiseLutWeightConfig
+from .codebook_quantized_tensor import GroupwiseLutQuantizedTensor
+
+__all__ = ["GroupwiseLutQuantizedTensor", "GroupwiseLutWeightConfig"]
