Spaces:

joaogante
/

medusa-maker

Paused

joaogante HF staff commited on Jan 3, 2024

Commit

4f6680e

1 Parent(s): d56736b

fix data path

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ print("Done")
 def create_medusa_heads(model_id: str):
     training_args = [
         "--model_name_or_path", model_id,
-        "--data_path", "data/ShareGPT_Vicuna_unfiltered/ShareGPT_V4.3_unfiltered_cleaned_split.json",
         "--bf16", "True",
         "--output_dir", "medusa_heads",
         "--num_train_epochs", "1",

 def create_medusa_heads(model_id: str):
     training_args = [
         "--model_name_or_path", model_id,
+        "--data_path", "data/ShareGPT_V4.3_unfiltered_cleaned_split.json",
         "--bf16", "True",
         "--output_dir", "medusa_heads",
         "--num_train_epochs", "1",