[Fix] Rollback opt model configs (#1213)

Fengzhe Zhou · web-flow · commit 0b50112dc19b · 2024-05-30T00:03:22.000+08:00
diff --git 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
@@ -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]
