Views
No views yet
1from transformers import GPT2LMHeadModel, GPT2Tokenizer
2
3model = GPT2LMHeadModel.from_pretrained("ViorikaAI/CalmaCatLM-2-mini")
4tokenizer = GPT2Tokenizer.from_pretrained("ViorikaAI/CalmaCatLM-2-mini")
5tokenizer.pad_token = tokenizer.eos_token
6
7inputs = tokenizer("Привет, как дела?", return_tensors="pt")
8outputs = model.generate(
9 **inputs,
10 max_new_tokens=80,
11 temperature=0.7,
12 top_k=50,
13 do_sample=True,
14 pad_token_id=tokenizer.eos_token_id,
15)
16print(tokenizer.decode(outputs[0], skip_special_tokens=True))