debugging: removing parallelize....

actualize-ae · Dec 31, 2023 · 52cc09a · 52cc09a
1 parent 05c9bdd
commit 52cc09a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/start.py b/start.py
@@ -65,7 +65,7 @@ def parse_arge():
 def run_model(model_id_or_path, tasks, num_fewshot=0):
     model_args = {
         "pretrained": model_id_or_path,  # required: taken from UI, no default value
-        "parallelize": True,
+        # "parallelize": True,
         "trust_remote_code": True
     }
     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):
     model_args = {
         "pretrained": model_id_path,  # required: taken from UI, no default value
         "peft": peft_model_id_or_path,
-        "parallelize": True,
+        # "parallelize": True,
         "trust_remote_code": True
     }
     model_args_str = make_model_args_str(model_args)