From f304eaa56de0516d2703299b87634430cf4e5c5a Mon Sep 17 00:00:00 2001
From: zhulinJulia24 <145004780+zhulinJulia24@users.noreply.github.com>
Date: Mon, 26 Feb 2024 15:52:57 +0800
Subject: [PATCH] add more models into daily test

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

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