add maxlen

bittersweet1999 · Dec 26, 2023 · f9b53a1 · f9b53a1
1 parent 82e9cb0
commit f9b53a1
Show file tree

Hide file tree

Showing 9 changed files with 18 additions and 18 deletions.
diff --git a/configs/models/judge_llm/auto_j/hf_autoj_bilingual_6b.py b/configs/models/judge_llm/auto_j/hf_autoj_bilingual_6b.py
@@ -7,7 +7,7 @@
 https://huggingface.co/GAIR/autoj-bilingual-6b
 '''
 
-models = dict(
+models = [dict(
         type=HuggingFaceCausalLM,
         abbr='autoj-bilingual-6b',
         path="GAIR/autoj-bilingual-6b",
@@ -21,4 +21,4 @@
         batch_size=8,
         model_kwargs=dict(device_map='auto', trust_remote_code=True),
         run_cfg=dict(num_gpus=1, num_procs=1),
-    )
+    )]
diff --git a/configs/models/judge_llm/auto_j/hf_autoj_eng_13b.py b/configs/models/judge_llm/auto_j/hf_autoj_eng_13b.py
@@ -1,7 +1,7 @@
 from opencompass.models import HuggingFaceCausalLM
 
 
-models = dict(
+models = [dict(
         type=HuggingFaceCausalLM,
         abbr='autoj-13b-GPTQ-4bits',
         path="GAIR/autoj-13b-GPTQ-4bits",
@@ -15,4 +15,4 @@
         batch_size=8,
         model_kwargs=dict(device_map='auto', trust_remote_code=True),
         run_cfg=dict(num_gpus=1, num_procs=1),
-    )
+    )]
diff --git a/configs/models/judge_llm/auto_j/hf_autoj_eng_13b_4bit.py b/configs/models/judge_llm/auto_j/hf_autoj_eng_13b_4bit.py
@@ -6,7 +6,7 @@
 https://huggingface.co/GAIR/autoj-13b-GPTQ-4bits
 '''
 
-models = dict(
+models = [dict(
         type=HuggingFaceCausalLM,
         abbr='autoj-13b',
         path="GAIR/autoj-13b",
@@ -20,4 +20,4 @@
         batch_size=8,
         model_kwargs=dict(device_map='auto', trust_remote_code=True),
         run_cfg=dict(num_gpus=1, num_procs=1),
-    )
+    )]
diff --git a/configs/models/judge_llm/auto_j/hf_autoj_scen_classifier.py b/configs/models/judge_llm/auto_j/hf_autoj_scen_classifier.py
@@ -1,7 +1,7 @@
 from opencompass.models import HuggingFaceCausalLM
 
 
-models = dict(
+models = [dict(
         type=HuggingFaceCausalLM,
         abbr='autoj-scenario-classifier',
         path="GAIR/autoj-scenario-classifier",
@@ -15,4 +15,4 @@
         batch_size=8,
         model_kwargs=dict(device_map='auto', trust_remote_code=True),
         run_cfg=dict(num_gpus=1, num_procs=1),
-    )
+    )]
diff --git a/configs/models/judge_llm/judgelm/hf_judgelm_13b_v1.py b/configs/models/judge_llm/judgelm/hf_judgelm_13b_v1.py
@@ -1,7 +1,7 @@
 from opencompass.models import HuggingFaceCausalLM
 
 
-models = dict(
+models = [dict(
         type=HuggingFaceCausalLM,
         abbr='judgelm-13b-v1-hf',
         path="BAAI/JudgeLM-13B-v1.0",
@@ -15,4 +15,4 @@
         batch_size=8,
         model_kwargs=dict(device_map='auto', trust_remote_code=True),
         run_cfg=dict(num_gpus=1, num_procs=1),
-    )
+    )]
diff --git a/configs/models/judge_llm/judgelm/hf_judgelm_33b_v1.py b/configs/models/judge_llm/judgelm/hf_judgelm_33b_v1.py
@@ -1,7 +1,7 @@
 from opencompass.models import HuggingFaceCausalLM
 
 
-models = dict(
+models = [dict(
         type=HuggingFaceCausalLM,
         abbr='judgelm-33b-v1-hf',
         path="BAAI/JudgeLM-33B-v1.0",
@@ -15,4 +15,4 @@
         batch_size=8,
         model_kwargs=dict(device_map='auto', trust_remote_code=True),
         run_cfg=dict(num_gpus=1, num_procs=1),
-    )
+    )]
diff --git a/configs/models/judge_llm/judgelm/hf_judgelm_7b_v1.py b/configs/models/judge_llm/judgelm/hf_judgelm_7b_v1.py
@@ -1,7 +1,7 @@
 from opencompass.models import HuggingFaceCausalLM
 
 
-models = dict(
+models = [dict(
         type=HuggingFaceCausalLM,
         abbr='judgelm-7b-v1-hf',
         path="BAAI/JudgeLM-7B-v1.0",
@@ -15,4 +15,4 @@
         batch_size=8,
         model_kwargs=dict(device_map='auto', trust_remote_code=True),
         run_cfg=dict(num_gpus=1, num_procs=1),
-    )
+    )]
diff --git a/configs/models/judge_llm/pandalm/hf_alpaca_pandalm_7b_v1.py b/configs/models/judge_llm/pandalm/hf_alpaca_pandalm_7b_v1.py
@@ -1,7 +1,7 @@
 from opencompass.models import HuggingFaceCausalLM
 
 
-models = dict(
+models = [dict(
         type=HuggingFaceCausalLM,
         abbr='alpaca-pandalm-7b-v1-hf',
         path="WeOpenML/PandaLM-Alpaca-7B-v1",
@@ -15,4 +15,4 @@
         batch_size=8,
         model_kwargs=dict(device_map='auto', trust_remote_code=True),
         run_cfg=dict(num_gpus=1, num_procs=1),
-    )
+    )]
diff --git a/configs/models/judge_llm/pandalm/hf_pandalm_7b_v1.py b/configs/models/judge_llm/pandalm/hf_pandalm_7b_v1.py
@@ -1,7 +1,7 @@
 from opencompass.models import HuggingFaceCausalLM
 
 
-models = dict(
+models = [dict(
         type=HuggingFaceCausalLM,
         abbr='pandalm-7b-v1-hf',
         path="WeOpenML/PandaLM-7B-v1",
@@ -15,4 +15,4 @@
         batch_size=8,
         model_kwargs=dict(device_map='auto', trust_remote_code=True),
         run_cfg=dict(num_gpus=1, num_procs=1),
-    )
+    )]