Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3# 加载模型
4model = AutoModelForCausalLM.from_pretrained("eason668/91395412-6602-4e54-a6a8-4dcca5b3bb04")
5tokenizer = AutoTokenizer.from_pretrained("eason668/91395412-6602-4e54-a6a8-4dcca5b3bb04")
6
7# 使用模型
8inputs = tokenizer("你的输入文本", return_tensors="pt")
9outputs = model.generate(**inputs, max_length=100)
10print(tokenizer.decode(outputs[0], skip_special_tokens=True))