github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix lint error

pull/710/head
zhulinJulia24 2024-02-27 14:01:13 +08:00 committed by GitHub
parent dfc1b09611
commit b073d840de
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_hf_model.py
View File

@ -1,6 +1,5 @@
import pytest import pytest
import torch, auto_gptq import torch
#from auto_gptq.modeling import BaseGPTQForCausalLM
from PIL import Image from PIL import Image
from transformers import AutoModel, AutoModelForCausalLM, AutoTokenizer from transformers import AutoModel, AutoModelForCausalLM, AutoTokenizer