noobmldude
commited on
Commit
•
63d9625
1
Parent(s):
115a6a3
Fix model type in example
Browse filesUse `gpt2` scripts instead of `bert` as model is GPT2 based.
README.md
CHANGED
@@ -79,7 +79,7 @@ you need to tell python where to find the clone of Megatron-LM, e.g.:
|
|
79 |
```
|
80 |
cd /tmp
|
81 |
git clone https://github.com/NVIDIA/Megatron-LM
|
82 |
-
PYTHONPATH=/tmp/Megatron-LM python src/transformers/models/
|
83 |
```
|
84 |
Or, if you already have it cloned elsewhere, simply adjust the path to the existing path.
|
85 |
|
|
|
79 |
```
|
80 |
cd /tmp
|
81 |
git clone https://github.com/NVIDIA/Megatron-LM
|
82 |
+
PYTHONPATH=/tmp/Megatron-LM python src/transformers/models/megatron_gpt2/convert_megatron_gpt2_checkpoint.py ...
|
83 |
```
|
84 |
Or, if you already have it cloned elsewhere, simply adjust the path to the existing path.
|
85 |
|