From 592a9113557ca19e4458a136db58340c3e08fcfc Mon Sep 17 00:00:00 2001 From: Rahul Tuli Date: Thu, 23 May 2024 10:17:31 -0400 Subject: [PATCH] Remove src from import --- src/sparseml/modifiers/quantization/gptq/pytorch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/sparseml/modifiers/quantization/gptq/pytorch.py b/src/sparseml/modifiers/quantization/gptq/pytorch.py index 1bea6224489..a3f36efb5b9 100644 --- a/src/sparseml/modifiers/quantization/gptq/pytorch.py +++ b/src/sparseml/modifiers/quantization/gptq/pytorch.py @@ -22,7 +22,7 @@ from sparseml.modifiers.quantization.gptq.base import GPTQModifier from sparseml.modifiers.utils.layer_compressor import LayerCompressor from sparseml.modifiers.utils.pytorch_helpers import run_calibration_forward -from src.sparseml.modifiers.quantization.gptq.utils.gptq_wrapper import GPTQWrapper +from sparseml.modifiers.quantization.gptq.utils.gptq_wrapper import GPTQWrapper __all__ = ["GPTQModifierPyTorch"]