fix IDE warning marlin not in __all__ (ModelCloud#920)

ZX-ModelCloud · Dec 19, 2024 · 71068ad · 71068ad
1 parent 5b8ea70
commit 71068ad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gptqmodel/nn_modules/qlinear/marlin.py b/gptqmodel/nn_modules/qlinear/marlin.py
@@ -443,4 +443,4 @@ def dequantize_weight(layer):
 
     return unpacked_qweight.T, unpacked_qzeros
 
-__all__ = ["dequantize_weight"]
+__all__ = ["MarlinQuantLinear", "dequantize_weight"]