From 9522572f2cc6a030ef4a0d59806c572156883a3e Mon Sep 17 00:00:00 2001
From: tooyosi <oluwatoyosi.oyegoke@gmail.com>
Date: Wed, 27 May 2026 15:56:53 +0100
Subject: [PATCH] add huggingFace env vars to batch env

---
 bajor/batch/train_finetuning.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/bajor/batch/train_finetuning.py b/bajor/batch/train_finetuning.py
index 0128185..9c4c16e 100644
--- a/bajor/batch/train_finetuning.py
+++ b/bajor/batch/train_finetuning.py
@@ -87,6 +87,13 @@ def create_batch_job(job_id, manifest_container_path, pool_id, options: JobOptio
             batchmodels.EnvironmentSetting(
                 name='WANDB_API_KEY',
                 value=os.getenv('WANDB_API_KEY', '')),
+            # set the Hugging Face env vars
+            batchmodels.EnvironmentSetting(
+                name='HF_REPO_ID',
+                value=os.getenv('HF_REPO_ID', '')),
+            batchmodels.EnvironmentSetting(
+                name='HF_TOKEN',
+                value=os.getenv('HF_TOKEN', '')),
             # set the zoobot saved model checkpoint file path
             batchmodels.EnvironmentSetting(
                 name='ZOOBOT_CHECKPOINT_TARGET',