Add __init__.py for group wise lut quantization package

szyszyzys · web-flow · commit 510e1b4510a4 · 2025-08-11T15:42:26.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"]
