Skip to content

Commit 52cc09a

Browse files
committed
debugging: removing parallelize....
1 parent 05c9bdd commit 52cc09a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

start.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ def parse_arge():
6565
def run_model(model_id_or_path, tasks, num_fewshot=0):
6666
model_args = {
6767
"pretrained": model_id_or_path, # required: taken from UI, no default value
68-
"parallelize": True,
68+
# "parallelize": True,
6969
"trust_remote_code": True
7070
}
7171
model_args_str = make_model_args_str(model_args)
@@ -83,7 +83,7 @@ def run_peft(model_id_path, peft_model_id_or_path, tasks, num_fewshot=0):
8383
model_args = {
8484
"pretrained": model_id_path, # required: taken from UI, no default value
8585
"peft": peft_model_id_or_path,
86-
"parallelize": True,
86+
# "parallelize": True,
8787
"trust_remote_code": True
8888
}
8989
model_args_str = make_model_args_str(model_args)

0 commit comments

Comments
 (0)