Remove unneeded commands in executable

GoogleCloudPlatform · Nov 5, 2024 · 376b7b8 · 376b7b8
1 parent 8f7a50d
commit 376b7b8
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/var/ramble/repos/builtin/applications/py-nemo/application.py b/var/ramble/repos/builtin/applications/py-nemo/application.py
@@ -35,8 +35,7 @@ class PyNemo(ExecutableApplication):
 
     executable(
         "pretraining_exec",
-        'bash -c "cd /opt/NeMo; git rev-parse HEAD; export PYTHONPATH=/opt/NeMo:\${PYTHONPATH}; '
-        "CUDA_VISIBLE_DEVICES={cuda_visible_devices} "
+        'bash -c "cd /opt/NeMo; git rev-parse HEAD; '
         "python3 -u /opt/NeMo/examples/nlp/language_modeling/megatron_gpt_pretraining.py "
         '--config-path={nemo_generated_config_path} --config-name={nemo_generated_config_name}"',
         use_mpi=True,