Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

fix path based configuration #456

Merged
merged 9 commits into from
May 21, 2024
Merged

fix path based configuration #456

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
c0db4d4
fix path based configuration
eaidova May 21, 2024
7559f31
Update llm_bench-python.yml
eaidova May 21, 2024
d864671
Update llm_bench-python.yml
eaidova May 21, 2024
378a6bd
Update requirements.txt
eaidova May 21, 2024
30a086a
Update llm_bench-python.yml
eaidova May 21, 2024
7050095
Update llm_bench-python.yml
eaidova May 21, 2024
fb97fff
Update .github/workflows/llm_bench-python.yml
eaidova May 21, 2024
4c06f7c
Update llm_bench-python.yml
eaidova May 21, 2024
4555a27
Update llm_bench-python.yml
eaidova May 21, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions .github/workflows/llm_bench-python.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ jobs:
run: |
python -m pip install --upgrade pip
python -m pip install flake8 pytest black
pip install -r ${{ env.LLM_BENCH_PYPATH }}/requirements.txt
GIT_CLONE_PROTECTION_ACTIVE=false pip install -r ${{ env.LLM_BENCH_PYPATH }}/requirements.txt
pip install openvino-nightly
- name: Lint with flake8
Expand Down Expand Up @@ -73,7 +73,7 @@ jobs:
python-version: 3.8
- name: Test stateful
run: |
python -m pip install -r llm_bench/python/requirements.txt
GIT_CLONE_PROTECTION_ACTIVE=false python -m pip install -r llm_bench/python/requirements.txt
python -m pip uninstall --yes openvino
python -m pip install openvino-nightly
python llm_bench/python/convert.py --model_id TinyLlama/TinyLlama-1.1B-Chat-v1.0 --output_dir . --stateful
Expand Down
6 changes: 3 additions & 3 deletions llm_bench/python/benchmark.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -174,7 +174,7 @@ def run_text_generation(input_text, num, model, tokenizer, args, iter_data_list,

def run_text_generation_benchmark(model_path, framework, device, args, num_iters):
model, tokenizer, pretrain_time, bench_hook = FW_UTILS[framework].create_text_gen_model(model_path, device, **args)
model_precision = utils.model_utils.get_model_precision(model_path.parents._parts)
model_precision = utils.model_utils.get_model_precision(model_path.parts)
iter_data_list = []
warmup_md5 = {}
input_text_list = utils.model_utils.get_prompts(args)
Expand Down Expand Up @@ -533,10 +533,10 @@ def main():
if args.report is not None or args.report_json is not None:
model_precision = ''
if framework == 'ov':
ir_conversion_frontend = utils.model_utils.get_ir_conversion_frontend(model_name, model_path.parents._parts)
ir_conversion_frontend = utils.model_utils.get_ir_conversion_frontend(model_name, model_path.parts)
if ir_conversion_frontend != '':
framework = framework + '(' + ir_conversion_frontend + ')'
model_precision = utils.model_utils.get_model_precision(model_path.parents._parts)
model_precision = utils.model_utils.get_model_precision(model_path.parts)
if args.report is not None:
utils.output_csv.write_result(
args.report,
Expand Down
2 changes: 1 addition & 1 deletion llm_bench/python/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ transformers>=4.33.0
diffusers>=0.22.0
#optimum is in dependency list of optimum-intel
git+https://github.com/huggingface/optimum-intel.git@8c2b787cc75a45ae4670d37970a5394eba90eedc#egg=optimum-intel
git+https://github.com/openvinotoolkit/nncf.git#egg=nncf
git+https://github.com/openvinotoolkit/nncf.git@develop#egg=nncf
packaging
psutil
timm
Expand Down
Loading