Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained("samz/minimind-pretrain")
4tokenizer = AutoTokenizer.from_pretrained("samz/minimind-pretrain")
5
6text = "今天天气真不错"
7inputs = tokenizer(text, return_tensors="pt")
8outputs = model.generate(**inputs, max_length=50)
9result = tokenizer.decode(outputs[0])
10print(result)