Update README.md
Browse files
README.md
CHANGED
|
@@ -44,4 +44,8 @@ inv_vocab = {idx: word for word, idx in vocab.items()}
|
|
| 44 |
# load model
|
| 45 |
model = MiniGPT(**json.load(open("config.json")))
|
| 46 |
model.load_state_dict(torch.load("mini_gpt_best.pt"))
|
| 47 |
-
model.eval()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 44 |
# load model
|
| 45 |
model = MiniGPT(**json.load(open("config.json")))
|
| 46 |
model.load_state_dict(torch.load("mini_gpt_best.pt"))
|
| 47 |
+
model.eval().
|
| 48 |
+
|
| 49 |
+
|
| 50 |
+
## Source code
|
| 51 |
+
https://github.com/m4vic/Transformer-101/blob/main/Decoder%20GPT%20style/Decoder.ipynb
|