Update README.md
Browse files
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
|
| 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'
|