Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2tok = AutoTokenizer.from_pretrained("gpt2")
3model = AutoModelForCausalLM.from_pretrained(
4 "itarutomy/llm_workshop_hands_on_gpt-model",
5 trust_remote_code=True
6)
7print(tok.decode(model.generate(**tok("こんにちは", return_tensors="pt"))[0]))