| from transformers import AutoModelForCausalLM, AutoTokenizer | |
| # Load GPT-2 small | |
| model = AutoModelForCausalLM.from_pretrained("gpt2") | |
| tokenizer = AutoTokenizer.from_pretrained("gpt2") |
| from transformers import AutoModelForCausalLM, AutoTokenizer | |
| # Load GPT-2 small | |
| model = AutoModelForCausalLM.from_pretrained("gpt2") | |
| tokenizer = AutoTokenizer.from_pretrained("gpt2") |