Skip to content

Commit 1498f79

Browse files
committed
Switch AI smoke test to direct CPU adapter run
1 parent 527cc80 commit 1498f79

1 file changed

Lines changed: 19 additions & 6 deletions

File tree

.github/workflows/ai_trading_smoke.yml

Lines changed: 19 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -6,15 +6,18 @@ on:
66
jobs:
77
ai-smoke:
88
runs-on: ubuntu-latest
9-
timeout-minutes: 30
9+
timeout-minutes: 45
1010
env:
1111
PYTHONUNBUFFERED: "1"
12-
TRAINED_MODEL_INFERENCE_URL: ${{ secrets.TRAINED_MODEL_INFERENCE_URL }}
13-
TRAINED_MODEL_API_KEY: ${{ secrets.TRAINED_MODEL_API_KEY }}
12+
MODAL_TOKEN_ID: ${{ secrets.MODAL_TOKEN_ID }}
13+
MODAL_TOKEN_SECRET: ${{ secrets.MODAL_TOKEN_SECRET }}
1414
TWELVEDATA_API_KEYS: ${{ secrets.TWELVEDATA_API_KEYS }}
1515
ALPHAVANTAGE_API_KEYS: ${{ secrets.ALPHAVANTAGE_API_KEYS }}
1616
NVIDIA_API_KEY: ${{ secrets.NVIDIA_API_KEY }}
1717
AI_SMOKE_TICKERS: "AAPL"
18+
TRAINED_MODEL_BASE_MODEL: "Qwen/Qwen2.5-7B-Instruct"
19+
TRAINED_MODEL_ADAPTER_PATH: "_smoke_artifacts/lora_solid_adapter"
20+
HF_HUB_DISABLE_TELEMETRY: "1"
1821
steps:
1922
- name: Checkout
2023
uses: actions/checkout@v4
@@ -25,15 +28,25 @@ jobs:
2528
python-version: "3.11"
2629

2730
- name: Install dependencies
28-
run: pip install -r requirements.txt
31+
run: |
32+
pip install -r requirements.txt
33+
pip install modal
34+
pip install torch==2.4.1 --index-url https://download.pytorch.org/whl/cpu
35+
pip install "transformers>=4.46.0" "peft>=0.13.2" "accelerate>=1.0.1" "sentencepiece>=0.2.0"
36+
37+
- name: Fetch trained adapter from Modal volume
38+
run: |
39+
mkdir -p _smoke_artifacts/lora_solid_adapter
40+
modal volume get train-once-artifacts /lora_solid_adapter/adapter_model.safetensors _smoke_artifacts/lora_solid_adapter/adapter_model.safetensors
41+
modal volume get train-once-artifacts /lora_solid_adapter/adapter_config.json _smoke_artifacts/lora_solid_adapter/adapter_config.json
2942
3043
- name: Run AI-only smoke test
31-
run: python run_ai_trading_smoke.py
44+
run: python run_ai_trading_smoke_direct.py
3245

3346
- name: Upload AI smoke artifacts
3447
if: always()
3548
uses: actions/upload-artifact@v4
3649
with:
3750
name: ai-trading-smoke
38-
path: results/ai_smoke_*.json
51+
path: results/ai_smoke_direct_*.json
3952
retention-days: 7

0 commit comments

Comments
 (0)