mirror of
https://github.com/lightbroker/llmsecops-research.git
synced 2026-04-30 07:17:54 +02:00
better handling for env var
This commit is contained in:
@@ -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")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user