Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2tok = AutoTokenizer.from_pretrained("Teto59/gpt2-finetuned-ja")
3mdl = AutoModelForCausalLM.from_pretrained("Teto59/gpt2-finetuned-ja")
4out = mdl.generate(**tok("こんにちは", return_tensors="pt"), max_new_tokens=30)
5print(tok.decode(out[0], skip_special_tokens=True))