Skip to content

Commit f34bf8b

Browse files
committed
typo
1 parent 54445e0 commit f34bf8b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.github/workflows/pull.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -859,7 +859,7 @@ jobs:
859859
python torchchat.py generate stories15M --temperature 0 --prompt "${PRMT}"
860860
861861
862-
for dtype in fp32, fp16, bf16; do
862+
for dtype in fp32 fp16 bf16; do
863863
echo "Testing export + runner with dtype=$dtype"
864864
python torchchat.py export stories15M --dtype $dtype --output-pte-path ./model.pte
865865
./cmake-out/et_run ./model.pte -z ./tokenizer.bin -t 0 -i "${PRMT}"
@@ -914,7 +914,7 @@ jobs:
914914
python torchchat.py generate --checkpoint-path ${MODEL_DIR}/stories15M.pt --temperature 0 --prompt "${PROMPT}"
915915
916916
917-
for dtype in fp32, fp16, bf16; do
917+
for dtype in fp32 fp16 bf16; do
918918
echo "Running export + runner with dtype=$dtype"
919919
python torchchat.py export --checkpoint-path ${MODEL_DIR}/stories15M.pt --dtype $dtype --output-dso-path /tmp/model.so
920920
./cmake-out/aoti_run /tmp/model.so -z ${MODEL_DIR}/tokenizer.bin -i "${PROMPT}"

0 commit comments

Comments
 (0)