From 3b622578efdbc70638240b8bdaf5a0a199940e7f Mon Sep 17 00:00:00 2001
From: David Hall <dlwh@stanford.edu>
Date: Sat, 7 Sep 2024 22:30:49 -0700
Subject: [PATCH] HF TOKEN

---
 .github/workflows/launch_small_fast.yaml | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/.github/workflows/launch_small_fast.yaml b/.github/workflows/launch_small_fast.yaml
index 5288bcb3c..46b7253d5 100644
--- a/.github/workflows/launch_small_fast.yaml
+++ b/.github/workflows/launch_small_fast.yaml
@@ -48,6 +48,16 @@ jobs:
           export TPU_NAME=small-fast-${{ github.run_id }}
           export WANDB_API_KEY=${{ secrets.WANDB_API_KEY }}
           export RUN_ID=small_fast_${{ github.run_id }}
+          export HF_TOKEN=${{ secrets.HF_TOKEN }}
+          
+          cat > .config <<EOF
+          env:
+              WANDB_API_KEY: ${{ secrets.WANDB_API_KEY }}
+              WANDB_ENTITY: stanford-mercury
+              WANDB_PROJECT: levanter
+              HF_TOKEN: ${{ secrets.HF_TOKEN }}
+          EOF
+          
           python infra/launch.py --foreground --tpu_name ${TPU_NAME} --run_id $RUN_ID --zone ${TPU_ZONE} --tpu_type ${TPU_TYPE} --preemptible -- \
              python -m levanter.main.train_lm \
                --config_path config/llama_small_fast.yaml \