Spaces:

joaogante
/

medusa-maker

Paused

joaogante HF staff commited on Jan 3

Commit

256858e

•

1 Parent(s): f5b4b7c

arg

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,8 +19,7 @@ def create_medusa_heads(model_id: str):
     parser = distributed_run.get_args_parser()
     args = parser.parse_args([
         "--nproc_per_node", "4",
-        "training_script", "medusa/medusa/train/train.py",
-        "training_script_args",
         "--model_name_or_path", model_id,
         "--data_path", "data/ShareGPT_V4.3_unfiltered_cleaned_split.json",
         "--bf16", "True",

     parser = distributed_run.get_args_parser()
     args = parser.parse_args([
         "--nproc_per_node", "4",
+        "medusa/medusa/train/train.py",
         "--model_name_or_path", model_id,
         "--data_path", "data/ShareGPT_V4.3_unfiltered_cleaned_split.json",
         "--bf16", "True",