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