Skip to content

Commit

Permalink
Test pytest command
Browse files Browse the repository at this point in the history
  • Loading branch information
archana-ramalingam committed Oct 28, 2024
1 parent c3aa964 commit 7d277d3
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions .github/workflows/ci_eval.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -67,11 +67,7 @@ jobs:
iree-runtime \
"numpy<2.0"
- name: Run perplexity test with vmfb
run: pytest -n 8 -v -s sharktank/tests/evaluate/perplexity_vmfb_test.py --longrun \
--iree-device='hip://7' \
--iree-hip-target='gfx942' \
--llama3-8b-f16-model-path=/data/llama-3.1/8b/llama8b_f16.irpa \
--llama3-8b-tokenizer-path=/data/llama-3.1/8b/tokenizer_config.json
run: pytest -n 8 -v -s sharktank/tests/evaluate/perplexity_vmfb_test.py --longrun --iree-device='hip://7' --iree-hip-target='gfx942' --llama3-8b-f16-model-path=/data/llama-3.1/8b/llama8b_f16.irpa --llama3-8b-tokenizer-path=/data/llama-3.1/8b/tokenizer_config.json

# test_perplexity_torch:
# timeout-minutes: 1000
Expand Down

0 comments on commit 7d277d3

Please sign in to comment.