[Fix] Rollback opt model configs (#1213)

open-compass · May 29, 2024 · 0b50112 · 0b50112
1 parent d656e81
commit 0b50112
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 20 deletions.
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]