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
models = [
dict(
type=HuggingFaceBaseModel,
abbr='opt-125m-hf',
path='facebook/opt-125m',
max_out_len=1024,
batch_size=64,
run_cfg=dict(num_gpus=1),
)
]
opt125m = dict(
type=HuggingFaceBaseModel,
abbr='opt-125m-hf',
path='facebook/opt-125m',
max_out_len=1024,
batch_size=64,
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
models = [
dict(
type=HuggingFaceBaseModel,
abbr='opt-350m-hf',
path='facebook/opt-350m',
max_out_len=1024,
batch_size=32,
run_cfg=dict(num_gpus=1),
)
]
opt350m = dict(
type=HuggingFaceBaseModel,
abbr='opt-350m-hf',
path='facebook/opt-350m',
max_out_len=1024,
batch_size=32,
run_cfg=dict(num_gpus=1),
)
models = [opt350m]