ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Update all-in-one benchmark readme (#9618)

Browse source
This commit is contained in:
Yuwen Hu 2023-12-07 10:32:09 +08:00 committed by GitHub
parent 0327169b50
commit 48b85593b3
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
python/llm/dev/benchmark/all-in-one/README.md
View file

@ -21,10 +21,12 @@ repo_id:
- 'THUDM/chatglm-6b' - 'THUDM/chatglm-6b'
- 'THUDM/chatglm2-6b' - 'THUDM/chatglm2-6b'
- 'meta-llama/Llama-2-7b-chat-hf' - 'meta-llama/Llama-2-7b-chat-hf'
# - 'liuhaotian/llava-v1.5-7b' # requires a LLAVA_REPO_DIR env variables pointing to the llava dir; added only for gpu win related test_api now
local_model_hub: 'path to your local model hub' local_model_hub: 'path to your local model hub'
warm_up: 1 warm_up: 1
num_trials: 3 num_trials: 3
num_beams: 1 # default to greedy search num_beams: 1 # default to greedy search
low_bit: 'sym_int4' # default to use 'sym_int4' (i.e. symmetric int4)
in_out_pairs: in_out_pairs:
- '32-32' - '32-32'
- '1024-128' - '1024-128'
@ -36,6 +38,9 @@ test_api:
# - "ipex_fp16_gpu" # on Intel GPU # - "ipex_fp16_gpu" # on Intel GPU
# - "transformer_int4_gpu" # on Intel GPU # - "transformer_int4_gpu" # on Intel GPU
# - "optimize_model_gpu" # on Intel GPU # - "optimize_model_gpu" # on Intel GPU
# - "deepspeed_transformer_int4_cpu" # on Intel SPR Server
# - "transformer_int4_gpu_win" # on Intel GPU for Windows (catch GPU peak memory)
cpu_embedding: False # whether put embedding to CPU (only avaiable now for gpu win related test_api)
``` ```
## Run ## Run