joaogante HF staff commited on
Commit
4f6680e
1 Parent(s): d56736b

fix data path

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ print("Done")
17
  def create_medusa_heads(model_id: str):
18
  training_args = [
19
  "--model_name_or_path", model_id,
20
- "--data_path", "data/ShareGPT_Vicuna_unfiltered/ShareGPT_V4.3_unfiltered_cleaned_split.json",
21
  "--bf16", "True",
22
  "--output_dir", "medusa_heads",
23
  "--num_train_epochs", "1",
 
17
  def create_medusa_heads(model_id: str):
18
  training_args = [
19
  "--model_name_or_path", model_id,
20
+ "--data_path", "data/ShareGPT_V4.3_unfiltered_cleaned_split.json",
21
  "--bf16", "True",
22
  "--output_dir", "medusa_heads",
23
  "--num_train_epochs", "1",