Skip to content

Commit

Permalink
[Update] Update requirement and deepseek configurations (#1764)
Browse files Browse the repository at this point in the history
  • Loading branch information
MaiziXiao authored Dec 17, 2024
1 parent 5c8e91f commit eadbdcb
Show file tree
Hide file tree
Showing 3 changed files with 23 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@
type=TurboMindModelwithChatTemplate,
abbr='deepseek-v2_5-turbomind',
path='deepseek-ai/DeepSeek-V2.5',
backend='pytorch',
engine_config=dict(
session_len=7168,
max_batch_size=4,
Expand Down
21 changes: 21 additions & 0 deletions opencompass/configs/models/deepseek/lmdeploy_deepseek_v2_5_1210.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
from opencompass.models import TurboMindModelwithChatTemplate

models = [
dict(
type=TurboMindModelwithChatTemplate,
abbr='deepseek-v2_5-1210-turbomind',
path='deepseek-ai/DeepSeek-V2.5-1210',
backend='pytorch',
engine_config=dict(
session_len=7168,
max_batch_size=4,
tp=8,
cache_max_entry_count=0.7,
),
gen_config=dict(top_k=1, temperature=1e-6, top_p=0.9),
max_seq_len=7168,
max_out_len=2048,
batch_size=4,
run_cfg=dict(num_gpus=8),
)
]
1 change: 1 addition & 0 deletions requirements/runtime.txt
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@ func_timeout
fuzzywuzzy
gradio-client
h5py
httpx==0.27.2
huggingface_hub<=0.24.7
immutabledict
importlib-metadata
Expand Down

0 comments on commit eadbdcb

Please sign in to comment.