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