From 0b50112dc19b5b06e32e41b6d7a3ba597ea516e8 Mon Sep 17 00:00:00 2001
From: Fengzhe Zhou <zfz-960727@163.com>
Date: Thu, 30 May 2024 00:03:22 +0800
Subject: [PATCH] [Fix] Rollback opt model configs (#1213)

---
 configs/models/opt/hf_opt_125m.py | 20 ++++++++++----------
 configs/models/opt/hf_opt_350m.py | 20 ++++++++++----------
 2 files changed, 20 insertions(+), 20 deletions(-)

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]