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

LLM: add rwkv 5 (#10048)

Browse source
This commit is contained in:
Jin Qiao 2024-01-31 15:54:55 +08:00 committed by GitHub
parent a9018a0e95
commit 8c8fc148c9
4 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
python/llm/test/benchmark/igpu-perf/1024-128.yaml
View file

@ -13,6 +13,7 @@ repo_id:
- 'mosaicml/mpt-7b-chat' - 'mosaicml/mpt-7b-chat'
- 'liuhaotian/llava-v1.5-7b' - 'liuhaotian/llava-v1.5-7b'
- 'RWKV/rwkv-4-world-7b' - 'RWKV/rwkv-4-world-7b'
- 'RWKV/rwkv-5-world-7b'
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

1
python/llm/test/benchmark/igpu-perf/2048-256.yaml
View file

@ -13,6 +13,7 @@ repo_id:
- 'mosaicml/mpt-7b-chat' - 'mosaicml/mpt-7b-chat'
- 'liuhaotian/llava-v1.5-7b' - 'liuhaotian/llava-v1.5-7b'
- 'RWKV/rwkv-4-world-7b' - 'RWKV/rwkv-4-world-7b'
- 'RWKV/rwkv-5-world-7b'
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

1
python/llm/test/benchmark/igpu-perf/32-32.yaml
View file

@ -13,6 +13,7 @@ repo_id:
- 'mosaicml/mpt-7b-chat' - 'mosaicml/mpt-7b-chat'
- 'liuhaotian/llava-v1.5-7b' - 'liuhaotian/llava-v1.5-7b'
- 'RWKV/rwkv-4-world-7b' - 'RWKV/rwkv-4-world-7b'
- 'RWKV/rwkv-5-world-7b'
local_model_hub: 'path to your local model hub' local_model_hub: 'path to your local model hub'
warm_up: 3 warm_up: 3
num_trials: 5 num_trials: 5

1
python/llm/test/benchmark/igpu-perf/32-512.yaml
View file

@ -13,6 +13,7 @@ repo_id:
- 'mosaicml/mpt-7b-chat' - 'mosaicml/mpt-7b-chat'
- 'liuhaotian/llava-v1.5-7b' - 'liuhaotian/llava-v1.5-7b'
- 'RWKV/rwkv-4-world-7b' - 'RWKV/rwkv-4-world-7b'
- 'RWKV/rwkv-5-world-7b'
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