Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3# 加载模型和 tokenizer
4model_name = "Longz1003/chatglm_lora_mimeng" # 替换为你的模型路径
5model = AutoModelForCausalLM.from_pretrained(model_name)
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7
8# 测试模型
9inputs = tokenizer("你好,今天的天气怎么样?", return_tensors="pt")
10outputs = model.generate(**inputs)
11print(tokenizer.decode(outputs[0], skip_special_tokens=True))
12
13torch>=1.10.0
14transformers>=4.30.0
15accelerate>=0.18.0
16datasets>=2.0.0
17scipy>=1.7.0
18sentencepiece>=0.1.96
19pytorch-lightning>=1.5.0
20pyyaml>=5.4.1
21tqdm>=4.62.0
22requests>=2.26.0
23huggingface_hub>=0.12.0
24tensorboard>=2.6.0
25scikit-learn>=0.24.2