jiangzhs/OpenCompass
1
0
Fork 0
mirror of https://github.com/open-compass/opencompass.git synced 2025-05-30 16:03:24 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Bump] Bump version to 0.3.7 (#1733)

Browse Source
This commit is contained in:
Linchen Xiao 2024-12-03 19:34:57 +08:00 committed by GitHub
parent 98c4666d65
commit e2a290fd46
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 39 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
opencompass/__init__.py
View File

@ -1,8 +1,9 @@
__version__ = '0.3.6' __version__ = '0.3.7'
def _warn_about_config_migration(): def _warn_about_config_migration():
import warnings import warnings
warnings.warn( warnings.warn(
'Starting from v0.4.0, all AMOTIC configuration files currently ' 'Starting from v0.4.0, all AMOTIC configuration files currently '
'located in `./configs/datasets`, `./configs/models`, and ' 'located in `./configs/datasets`, `./configs/models`, and '
@ -10,7 +11,8 @@ def _warn_about_config_migration():
'`opencompass/configs/` package. Please update your configuration ' '`opencompass/configs/` package. Please update your configuration '
'file paths accordingly.', 'file paths accordingly.',
UserWarning, # Changed to UserWarning UserWarning, # Changed to UserWarning
stacklevel=2) stacklevel=2,
)
# Trigger the warning # Trigger the warning

2
opencompass/configs/models/deepseek/lmdeploy_deepseek_v2_lite.py
View File

@ -3,7 +3,7 @@ from opencompass.models import TurboMindModelwithChatTemplate
models = [ models = [
dict( dict(
type=TurboMindModelwithChatTemplate, type=TurboMindModelwithChatTemplate,
abbr='deepseek-v2_lite-turbomind', abbr='deepseek-v2_lite-chat-turbomind',
path='deepseek-ai/DeepSeek-V2-Lite-Chat', path='deepseek-ai/DeepSeek-V2-Lite-Chat',
engine_config=dict( engine_config=dict(
session_len=7168, session_len=7168,

17
opencompass/configs/models/gemma/lmdeploy_gemma_27b.py Normal file
View File

@ -0,0 +1,17 @@
from opencompass.models import TurboMindModel
models = [
dict(
type=TurboMindModel,
abbr='gemma-2-27b-turbomind',
path='google/gemma-2-27b',
engine_config=dict(session_len=16384, max_batch_size=16, tp=2),
gen_config=dict(
top_k=1, temperature=1e-6, top_p=0.9, max_new_tokens=4096
),
max_seq_len=16384,
max_out_len=4096,
batch_size=16,
run_cfg=dict(num_gpus=2),
)
]

17
opencompass/configs/models/gemma/lmdeploy_gemma_9b.py Normal file
View File

@ -0,0 +1,17 @@
from opencompass.models import TurboMindModel
models = [
dict(
type=TurboMindModel,
abbr='gemma-2-9b-turbomind',
path='google/gemma-2-9b',
engine_config=dict(session_len=16384, max_batch_size=16, tp=1),
gen_config=dict(
top_k=1, temperature=1e-6, top_p=0.9, max_new_tokens=4096
),
max_seq_len=16384,
max_out_len=4096,
batch_size=16,
run_cfg=dict(num_gpus=1),
)
]