Skip to content

Commit cf38639

Browse files
committed
Harden AI smoke workflow timeout and CPU settings
1 parent 1498f79 commit cf38639

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

.github/workflows/ai_trading_smoke.yml

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ on:
66
jobs:
77
ai-smoke:
88
runs-on: ubuntu-latest
9-
timeout-minutes: 45
9+
timeout-minutes: 90
1010
env:
1111
PYTHONUNBUFFERED: "1"
1212
MODAL_TOKEN_ID: ${{ secrets.MODAL_TOKEN_ID }}
@@ -18,6 +18,8 @@ jobs:
1818
TRAINED_MODEL_BASE_MODEL: "Qwen/Qwen2.5-7B-Instruct"
1919
TRAINED_MODEL_ADAPTER_PATH: "_smoke_artifacts/lora_solid_adapter"
2020
HF_HUB_DISABLE_TELEMETRY: "1"
21+
HF_HUB_ENABLE_HF_TRANSFER: "1"
22+
TRAINED_MODEL_CPU_THREADS: "4"
2123
steps:
2224
- name: Checkout
2325
uses: actions/checkout@v4
@@ -30,7 +32,7 @@ jobs:
3032
- name: Install dependencies
3133
run: |
3234
pip install -r requirements.txt
33-
pip install modal
35+
pip install modal hf_transfer
3436
pip install torch==2.4.1 --index-url https://download.pytorch.org/whl/cpu
3537
pip install "transformers>=4.46.0" "peft>=0.13.2" "accelerate>=1.0.1" "sentencepiece>=0.2.0"
3638

0 commit comments

Comments
 (0)