from transformers import AutoModel, AutoTokenizer

tokenizer = AutoTokenizer.from_pretrained("/home/yudong/work/VisualGLM-6B/THUDM/visualglm-6b", trust_remote_code=True, revision="")
model = AutoModel.from_pretrained("/home/yudong/work/VisualGLM-6B/THUDM/visualglm-6b", trust_remote_code=True, revision="").float()
model = model.eval()

# 将模型和令牌器设置为全局变量
global_model = model
global_tokenizer = tokenizer