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

[Fix] Rollback opt model configs (#1213)

Browse Source
This commit is contained in:
Fengzhe Zhou 2024-05-30 00:03:22 +08:00 committed by GitHub
parent d656e818f8
commit 0b50112dc1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 20 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
configs/models/opt/hf_opt_125m.py
View File

@ -1,12 +1,12 @@
from opencompass.models import HuggingFaceBaseModel from opencompass.models import HuggingFaceBaseModel
models = [ opt125m = dict(
dict( type=HuggingFaceBaseModel,
type=HuggingFaceBaseModel, abbr='opt-125m-hf',
abbr='opt-125m-hf', path='facebook/opt-125m',
path='facebook/opt-125m', max_out_len=1024,
max_out_len=1024, batch_size=64,
batch_size=64, run_cfg=dict(num_gpus=1),
run_cfg=dict(num_gpus=1), )
)
] models = [opt125m]

20
configs/models/opt/hf_opt_350m.py
View File

@ -1,12 +1,12 @@
from opencompass.models import HuggingFaceBaseModel from opencompass.models import HuggingFaceBaseModel
models = [ opt350m = dict(
dict( type=HuggingFaceBaseModel,
type=HuggingFaceBaseModel, abbr='opt-350m-hf',
abbr='opt-350m-hf', path='facebook/opt-350m',
path='facebook/opt-350m', max_out_len=1024,
max_out_len=1024, batch_size=32,
batch_size=32, run_cfg=dict(num_gpus=1),
run_cfg=dict(num_gpus=1), )
)
] models = [opt350m]