phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-01 16:24:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Clean up checkpoint directory setting

Browse source
This commit is contained in:
Connor Hack 2024-11-22 13:52:58 -08:00
parent d1d8f859e6
commit cbd69d06c3
1 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
.github/workflows/gha_workflow_llama_stack_tests.yml vendored
View file

@ -44,12 +44,12 @@ on:
model_override_3b:
description: 'Specify manual override for the <llama_3b> shorthand model'
required: false
default: "Llama-3.2-3B-Instruct"
default: "Llama3.2-3B-Instruct"
model_override_8b:
description: 'Specify manual override for the <llama_8b> shorthand model'
required: false
default: "Llama-3.1-8B-Instruct"
default: "Llama3.1-8B-Instruct"
env:
# ID used for each test's provider config
@ -68,8 +68,8 @@ env:
MODEL_ID: "${{ inputs.model_id || 'llama_3b' }}"
# Set the llama 3b / 8b override for models if desired, else use the default.
LLAMA_3B_OVERRIDE: "${{ inputs.model_override_3b || 'Llama-3.2-3B-Instruct' }}"
LLAMA_8B_OVERRIDE: "${{ inputs.model_override_8b || 'Llama-3.1-8B-Instruct' }}"
LLAMA_3B_OVERRIDE: "${{ inputs.model_override_3b || 'Llama3.2-3B-Instruct' }}"
LLAMA_8B_OVERRIDE: "${{ inputs.model_override_8b || 'Llama3.1-8B-Instruct' }}"
# Defines which directories in TESTS_PATH to exclude from the test loop
EXCLUDED_DIRS: "__pycache__"
@ -181,6 +181,7 @@ jobs:
- name: "Update checkpoint directory"
id: checkpoint_update
run: |
echo "Checkpoint directory: ${MODEL_CHECKPOINT_DIR}/$LLAMA_3B_OVERRIDE"
if ${MODEL_ID} == "llama_3b" && [ -d "${MODEL_CHECKPOINT_DIR}/$LLAMA_3B_OVERRIDE" ]; then
echo "MODEL_CHECKPOINT_DIR=${MODEL_CHECKPOINT_DIR}/$LLAMA_3B_OVERRIDE" >> "$GITHUB_ENV"
elif ${MODEL_ID} == "llama_8b" && [ -d "${MODEL_CHECKPOINT_DIR}/$LLAMA_8B_OVERRIDE" ]; then