diff --git a/modeling/inference_models/gptq_hf_torch/class.py b/modeling/inference_models/gptq_hf_torch/class.py index d942a539..499b2682 100644 --- a/modeling/inference_models/gptq_hf_torch/class.py +++ b/modeling/inference_models/gptq_hf_torch/class.py @@ -90,7 +90,6 @@ class model_backend(HFTorchInferenceModel): def _load(self, save_model: bool, initial_load: bool) -> None: try: - import hf_bleeding_edge from hf_bleeding_edge import AutoModelForCausalLM except ImportError: from transformers import AutoModelForCausalLM