From 83743ada8c0c48b457d2c2fde4063c35978d5121 Mon Sep 17 00:00:00 2001 From: wgzintel Date: Thu, 1 Aug 2024 16:46:56 +0800 Subject: [PATCH] change bloomz-560m to tiny-random-baichuan2 --- .github/workflows/llm_bench-python.yml | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) 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