Skip to content
This repository was archived by the owner on Sep 10, 2025. It is now read-only.

Commit ef74428

Browse files
authored
Update more-tests.yml
1 parent 89746a1 commit ef74428

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

.github/workflows/more-tests.yml

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -40,9 +40,10 @@ jobs:
4040
echo "::endgroup::"
4141
4242
echo "::group::Run inference"
43-
export MODEL_PATH=checkpoints/stories15M/stories15M.pt
44-
export MODEL_NAME=stories15M
4543
export MODEL_DIR=checkpoints/stories15M/
44+
export MODEL_PATH=${MODEL_DIR}/stories15M.pt
45+
export MODEL_NAME=stories15M
46+
4647
4748
for DTYPE in bfloat16 float16 float32; do
4849
###################################################################
@@ -118,9 +119,9 @@ jobs:
118119
echo "::endgroup::"
119120
120121
echo "::group::Run inference"
121-
export MODEL_PATH=checkpoints/stories15M/stories15M.pt
122+
export MODEL_DIR=checkpoints/stories15M/
123+
export MODEL_PATH=${MODEL_DIR}/stories15M.pt
122124
export MODEL_NAME=stories15M
123-
export MODEL_DIR=checkpoints/stories15M
124125
125126
./torchchat/utils/scripts/build_native.sh aoti
126127

0 commit comments

Comments
 (0)