[examples] Update examples using deprecated SKYPILOT_JOB_ID (#3344)

Update examples
skypilot-org · Mar 21, 2024 · cd8a8c1 · cd8a8c1
1 parent 25c7ef5
commit cd8a8c1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/examples/tensorflow_distributed/tf_distributed.yaml b/examples/tensorflow_distributed/tf_distributed.yaml
@@ -53,12 +53,12 @@ run: |
               'index': node_rank
           }
       }
-  with open(f'/tmp/{os.environ.get("SKYPILOT_JOB_ID")}', 'w') as f:
+  with open(f'/tmp/{os.environ.get("SKYPILOT_TASK_ID")}', 'w') as f:
       json.dump(tf_config, f)
   EOF
   
   # Read and set TF_CONFIG from file
-  export TF_CONFIG=$(cat /tmp/$SKYPILOT_JOB_ID)
+  export TF_CONFIG=$(cat /tmp/$SKYPILOT_TASK_ID)
   echo $TF_CONFIG
   
   # ======== Run the training script ========

diff --git a/llm/vicuna/train.yaml b/llm/vicuna/train.yaml
@@ -100,7 +100,7 @@ run: |
     --fsdp_transformer_layer_cls_to_wrap 'LlamaDecoderLayer' \
     --tf32 True \
     --model_max_length ${SEQ_LEN} \
-    --run_name $SKYPILOT_JOB_ID \
+    --run_name $SKYPILOT_TASK_ID \
     --gradient_checkpointing True \
     --lazy_preprocess True