Hugging Face's logo

nvidia
/
megatron-gpt2-345m

Model card Files Community
2
jdemouth commited on
Commit
89445bf
•
1 Parent(s): 65c6843

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -118,7 +118,7 @@ wget --content-disposition https://api.ngc.nvidia.com/v2/models/nvidia/megatron_
118
  python src/transformers/models/megatron_gpt2/convert_megatron_gpt2_checkpoint.py megatron_lm_345m_v0.0.zip
119
  ```
120
 
121
- As explained by Stas Bekman in [PR #14956](https://github.com/huggingface/transformers/pull/14956), if when running this conversion
122
  script and you're getting an exception:
123
  ```
124
  ModuleNotFoundError: No module named 'megatron.model.enums'
 
118
  python src/transformers/models/megatron_gpt2/convert_megatron_gpt2_checkpoint.py megatron_lm_345m_v0.0.zip
119
  ```
120
 
121
+ As explained in [PR #14956](https://github.com/huggingface/transformers/pull/14956), if when running this conversion
122
  script and you're getting an exception:
123
  ```
124
  ModuleNotFoundError: No module named 'megatron.model.enums'