fix env var extraction (#2043) [skip ci]

axolotl-ai-cloud · Nov 18, 2024 · a865d3d · a865d3d
1 parent 256ceda
commit a865d3d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/cloud-entrypoint.sh b/scripts/cloud-entrypoint.sh
@@ -2,7 +2,7 @@
 
 # Export specific ENV variables to /etc/rp_environment
 echo "Exporting environment variables..."
-printenv | grep -E '^RUNPOD_|^PATH=|^_=' | sed 's/^\(.*\)=\(.*\)$/export \1="\2"/' >> /etc/rp_environment
+printenv | grep -E '^HF_|^BNB_|^CUDA_|^NCCL_|^NV|^RUNPOD_|^PATH=|^_=' | sed 's/^\([^=]*\)=\(.*\)$/export \1="\2"/' | grep -v 'printenv' >> /etc/rp_environment
 echo 'source /etc/rp_environment' >> ~/.bashrc
 
 add_keys_to_authorized() {