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