diff --git a/.github/workflows/llm_bench-python.yml b/.github/workflows/llm_bench-python.yml index 1f59bc91c2..d9609ffa9d 100644 --- a/.github/workflows/llm_bench-python.yml +++ b/.github/workflows/llm_bench-python.yml @@ -55,15 +55,15 @@ jobs: with: name: llm.bench_diff path: llm.bench_diff.diff - - name: Test tiny-random-qwen of pytorch model on Linux + - name: Test native pytorch model on Linux run: | export GIT_LFS_SKIP_SMUDGE=0 - git clone https://huggingface.co/katuni4ka/tiny-random-qwen + git clone --depth 1 https://huggingface.co/katuni4ka/tiny-random-qwen python ./llm_bench/python/benchmark.py -m tiny-random-qwen -d cpu -n 1 -f pt - - name: Test bloomz-560m on Linux + - name: Test tiny-random-baichuan2 on Linux run: | - python ./llm_bench/python/convert.py --model_id bigscience/bloomz-560m --output_dir ./ov_models/bloomz-560m --precision FP16 - python ./llm_bench/python/benchmark.py -m ./ov_models/bloomz-560m/pytorch/dldt/FP16/ -d cpu -n 1 + python ./llm_bench/python/convert.py --model_id katuni4ka/tiny-random-baichuan2 --output_dir ./ov_models/tiny-random-baichuan2 --precision FP16 + python ./llm_bench/python/benchmark.py -m ./ov_models/tiny-random-baichuan2/pytorch/dldt/FP16/ -d cpu -n 1 - name: Test tiny-stable-diffusion on Linux run: | python ./llm_bench/python/convert.py --model_id segmind/tiny-sd --output_dir ./ov_models/tiny-sd --precision FP16