remove src. from imports

neuralmagic · Mar 26, 2024 · 7ddc3f0 · 7ddc3f0
1 parent 1b1567d
commit 7ddc3f0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/sparseml/transformers/utils/vllm_export_helpers.py b/src/sparseml/transformers/utils/vllm_export_helpers.py
@@ -23,6 +23,8 @@
 from torch import Tensor
 from transformers import PreTrainedModel, PreTrainedTokenizerBase
 
+from sparseml.transformers.utils.sparse_model import SparseAutoModelForCausalLM
+from sparseml.transformers.utils.sparse_tokenizer import SparseAutoTokenizer
 from sparseml.transformers.utils.transformations import (
     add_tensors,
     convert_fp32_to_fp16,
@@ -31,8 +33,6 @@
     transform_tensors,
 )
 from sparseml.utils import get_unique_dir_name
-from src.sparseml.transformers.utils.sparse_model import SparseAutoModelForCausalLM
-from src.sparseml.transformers.utils.sparse_tokenizer import SparseAutoTokenizer
 
 
 __all__ = [