Update Reference Container for Nemo FW

README.md

@@ -31,7 +31,7 @@ Under the NVIDIA Open Model License, NVIDIA confirms:
 
 ### Intended use
 
-Nemotron-4-340B-Base is a completion model intended for use in over 50+ natural and 40+ coding languages. It is compatible with [NVIDIA NeMo Framework](https://docs.nvidia.com/nemo-framework/index.html). For best performance on a given task, users are encouraged to customize the model using the NeMo Framework suite of customization tools including Parameter-Efficient Fine-Tuning (P-tuning, Adapters, LoRA, and more), and Model Alignment (SFT, SteerLM, RLHF, and more) using [NeMo-Aligner](https://github.com/NVIDIA/NeMo-Aligner).
+Nemotron-4-340B-Base is a completion model intended for use in over 50+ natural and 40+ coding languages. It is compatible with [NVIDIA NeMo Framework](https://docs.nvidia.com/nemo-framework/index.html). For best performance on a given task, users are encouraged to customize the model using the NeMo Framework suite of customization tools including Parameter-Efficient Fine-Tuning (P-tuning, Adapters, LoRA, and more), and Model Alignment (SFT, SteerLM, RLHF, and more) using [NeMo-Aligner](https://github.com/NVIDIA/NeMo-Aligner). Refer to the [documentation](https://docs.nvidia.com/nemo-framework/user-guide/latest/llms/nemotron/index.html) for examples.
 
 **Model Developer:** NVIDIA
 
@@ -105,7 +105,7 @@ print(response)
 ```
 
 
-2. Given this Python script, create a Bash script which spins up the inference server within the [NeMo container](https://catalog.ngc.nvidia.com/orgs/nvidia/containers/nemo) (```docker pull nvcr.io/nvidia/nemo:24.01.framework```) and calls the Python script ``call_server.py``. The Bash script ``nemo_inference.sh`` is as follows:
+2. Given this Python script, create a Bash script which spins up the inference server within the [NeMo container](https://catalog.ngc.nvidia.com/orgs/nvidia/containers/nemo) (```docker pull nvcr.io/nvidia/nemo:24.05```) and calls the Python script ``call_server.py``. The Bash script ``nemo_inference.sh`` is as follows:
 
 
 ```bash
@@ -171,7 +171,7 @@ RESULTS=<PATH_TO_YOUR_SCRIPTS_FOLDER>
 OUTFILE="${RESULTS}/slurm-%j-%n.out"
 ERRFILE="${RESULTS}/error-%j-%n.out"
 MODEL=<PATH_TO>/Nemotron-4-340B-Base
-CONTAINER="nvcr.io/nvidia/nemo:24.01.framework"
+CONTAINER="nvcr.io/nvidia/nemo:24.05"
 MOUNTS="--container-mounts=<PATH_TO_YOUR_SCRIPTS_FOLDER>:/scripts,MODEL:/model"
 
 read -r -d '' cmd <<EOF