better handling for env var

This commit is contained in:
Adam Wilson
2025-08-19 12:11:25 -06:00
parent ee9355e757
commit c7fa024dc5
2 changed files with 22 additions and 1 deletions
@@ -47,6 +47,19 @@ jobs:
shell: bash
run: pip install -r ${{ github.workspace }}/requirements.txt
- name: 'ensure HF_TOKEN'
shell: bash
run: |
echo "HF_TOKEN length: ${#HF_TOKEN}"
if [ -z "$HF_TOKEN" ]; then
echo "ERROR: HF_TOKEN is empty or not set"
exit 1
else
echo "HF_TOKEN is set (first 10 chars): ${HF_TOKEN:0:10}..."
fi
env:
HF_TOKEN: ${{ secrets.HF_TOKEN }}
- name: 'run text generation tests - ${{ inputs.model_display_name }} - range ${{ inputs.range_name }} batch ${{ matrix.batch }}'
shell: bash
env:
@@ -7,7 +7,15 @@ from src.text_generation.common.model_id import ModelId
from huggingface_hub import login
if "HF_TOKEN" in os.environ:
login(token=os.environ["HF_TOKEN"])
hf_token = os.environ.get("HF_TOKEN", "").strip()
if not hf_token:
raise ValueError("HF_TOKEN environment variable is empty or not found")
try:
login(token=hf_token)
print("Successfully authenticated with Hugging Face")
except Exception as e:
raise ValueError(f"Failed to authenticate with Hugging Face: {e}")
else:
raise ValueError("HF_TOKEN environment variable not found")