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