255
名前なし
2026/01/07 (水) 16:56:42
c194f@0c87e
from transformers import AutoTokenizer, AutoModelForCausalLM
model_name = "sshleifer/tiny-gpt2" # 軽いテスト用モデル
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name)
prompt = "こんにちは！調子どう？"
inputs = tokenizer(prompt, return_tensors="pt")
outputs = model.generate(**inputs, max_length=60)
print(tokenizer.decode(outputs[0]))
通報 ...