diff --git a/configs/models/opt/hf_opt_125m.py b/configs/models/opt/hf_opt_125m.py index ec0c68d3..b813d28e 100644 --- a/configs/models/opt/hf_opt_125m.py +++ b/configs/models/opt/hf_opt_125m.py @@ -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] diff --git a/configs/models/opt/hf_opt_350m.py b/configs/models/opt/hf_opt_350m.py index 6a25db0a..57aa609c 100644 --- a/configs/models/opt/hf_opt_350m.py +++ b/configs/models/opt/hf_opt_350m.py @@ -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]