Merge pull request #133 from NVIDIA/23.08.02

change container to 23.08.02
NVIDIA · Sep 19, 2023 · 561a251 · 561a251
2 parents d24f8ef + 2078c59
commit 561a251
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -1976,7 +1976,7 @@ launcher_scripts_path: ${auto_configurator_path}/../launcher_scripts
 fastertransformer_path: ${auto_configurator_path}/../FasterTransformer
 base_results_dir: ${auto_configurator_path}/results
 data_dir: ${launcher_scripts_path}/data
-training_container: nvcr.io/ea-bignlp/ga-participants/nemofw-training:23.08.01
+training_container: nvcr.io/ea-bignlp/ga-participants/nemofw-training:23.08.02
 container_mounts:
     - null
 wandb:  # Weights and Biases (W&B) logging.
@@ -5400,7 +5400,7 @@ VALID_DATA_PATH=/path/to/val_actor
 TEST_DATA_PATH=/path/to/test_actor
 
 NEMO_RLHF_DIR=/opt/nemo-rlhf
-CONTAINER="nvcr.io/ea-bignlp/ga-participants/nemofw-training:23.08.01"
+CONTAINER="nvcr.io/ea-bignlp/ga-participants/nemofw-training:23.08.02"
 
 mkdir -p $OUTPUT_DIR
 

diff --git a/auto_configurator/conf/config.yaml b/auto_configurator/conf/config.yaml
@@ -18,7 +18,7 @@ launcher_scripts_path: ${auto_configurator_path}/../launcher_scripts
 base_results_dir: ${auto_configurator_path}/results
 data_dir: ${launcher_scripts_path}/data
 
-training_container: nvcr.io/ea-bignlp/ga-participants/nemofw-training:23.08.01
+training_container: nvcr.io/ea-bignlp/ga-participants/nemofw-training:23.08.02
 container_mounts:
   - null
 

diff --git a/auto_configurator/tests/config_tests/test_main_config.py b/auto_configurator/tests/config_tests/test_main_config.py
@@ -26,7 +26,7 @@ def test_config(self):
         base_results_dir: ${auto_configurator_path}/results
         data_dir: ${launcher_scripts_path}/data
 
-        training_container: nvcr.io/ea-bignlp/ga-participants/nemofw-training:23.08.01
+        training_container: nvcr.io/ea-bignlp/ga-participants/nemofw-training:23.08.02
         container_mounts:
           - null
         

diff --git a/csp_tools/aws/Dockerfile b/csp_tools/aws/Dockerfile
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-FROM nvcr.io/ea-bignlp/nemofw-training:23.07-py3
+FROM nvcr.io/ea-bignlp/ga-participants/nemofw-training:23.08.02
 
 ARG EFA_INSTALLER_VERSION=latest
 ARG AWS_OFI_NCCL_VERSION=1.4.0-aws

diff --git a/launcher_scripts/conf/config.yaml b/launcher_scripts/conf/config.yaml
@@ -39,7 +39,7 @@ data_dir: ${launcher_scripts_path}/data  # Location to store and read the data.
 base_results_dir: ${launcher_scripts_path}/results  # Location to store the results, checkpoints and logs.
 container_mounts: # List of additional paths to mount to container. They will be mounted to same path.
   - null
-container: nvcr.io/ea-bignlp/ga-participants/nemofw-training:23.08.01
+container: nvcr.io/ea-bignlp/ga-participants/nemofw-training:23.08.02
 
 wandb_api_key_file: null  # File where the w&B api key is stored. Key must be on the first line.