nvidia
/

megatron-gpt2-345m

Model card Files Files and versions Community

Fix model type in example

#1

by noobmldude - opened Oct 12, 2023

base: refs/heads/main

←

from: refs/pr/1

Discussion Files changed

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -79,7 +79,7 @@ you need to tell python where to find the clone of Megatron-LM, e.g.:
 ```
 cd /tmp
 git clone https://github.com/NVIDIA/Megatron-LM
-PYTHONPATH=/tmp/Megatron-LM python src/transformers/models/megatron_bert/convert_megatron_bert_checkpoint.py ...
 ```
 Or, if you already have it cloned elsewhere, simply adjust the path to the existing path.

 ```
 cd /tmp
 git clone https://github.com/NVIDIA/Megatron-LM
+PYTHONPATH=/tmp/Megatron-LM python src/transformers/models/megatron_gpt2/convert_megatron_gpt2_checkpoint.py ...
 ```
 Or, if you already have it cloned elsewhere, simply adjust the path to the existing path.