From b073d840de8ebd1187a7ab4ee5dd536e29505087 Mon Sep 17 00:00:00 2001
From: zhulinJulia24 <145004780+zhulinJulia24@users.noreply.github.com>
Date: Tue, 27 Feb 2024 14:01:13 +0800
Subject: [PATCH] fix lint error

---
 tests/test_hf_model.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/tests/test_hf_model.py b/tests/test_hf_model.py
index a3f11b3..4bbdcd3 100644
--- a/tests/test_hf_model.py
+++ b/tests/test_hf_model.py
@@ -1,6 +1,5 @@
 import pytest
-import torch, auto_gptq
-#from auto_gptq.modeling import BaseGPTQForCausalLM
+import torch
 from PIL import Image
 from transformers import AutoModel, AutoModelForCausalLM, AutoTokenizer