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 |
|