jojo-ai-mst commited on
Commit
247ebbb
1 Parent(s): e2f5e05

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -65,7 +65,7 @@ print(outputs)
65
 
66
  ```python
67
  import torch
68
- from transformers import AutoTokenizer, AutoModelForCausalLM
69
 
70
  model = GPT2LMHeadModel.from_pretrained("jojo-ai-mst/MyanmarGPT")
71
  tokenizer = GPT2Tokenizer.from_pretrained("jojo-ai-mst/MyanmarGPT")
 
65
 
66
  ```python
67
  import torch
68
+ from transformers import GPT2Tokenizer, GPT2LMHeadModel
69
 
70
  model = GPT2LMHeadModel.from_pretrained("jojo-ai-mst/MyanmarGPT")
71
  tokenizer = GPT2Tokenizer.from_pretrained("jojo-ai-mst/MyanmarGPT")