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