NeMo
PyTorch
text generation
causal-lm
okuchaiev commited on
Commit
78609cb
1 Parent(s): 18c8a57

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -96,14 +96,14 @@ Note: You will need NVIDIA Ampere or Hopper GPUs to work with this model.
96
  You will need to install NVIDIA Apex and NeMo.
97
 
98
  ```
99
- git clone https://github.com/ericharper/apex.git
100
  cd apex
101
- git checkout nm_v1.11.0
102
  pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" --global-option="--fast_layer_norm" --global-option="--distributed_adam" --global-option="--deprecated_fused_adam" ./
103
  ```
104
 
105
  ```
106
- pip install nemo_toolkit['nlp']==1.11.0
107
  ```
108
 
109
  Alternatively, you can use NeMo Megatron training docker container with all dependencies pre-installed.
 
96
  You will need to install NVIDIA Apex and NeMo.
97
 
98
  ```
99
+ git clone https://github.com/NVIDIA/apex.git
100
  cd apex
101
+ git checkout 03c9d80ed54c0eaa5b581bf42ceca3162f085327
102
  pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" --global-option="--fast_layer_norm" --global-option="--distributed_adam" --global-option="--deprecated_fused_adam" ./
103
  ```
104
 
105
  ```
106
+ pip install nemo_toolkit['nlp']==1.17.0
107
  ```
108
 
109
  Alternatively, you can use NeMo Megatron training docker container with all dependencies pre-installed.