Hugging Face's logo

Spaces:
joaogante
/
medusa-maker
Paused

App Files Community
1
joaogante HF staff commited on
Commit
6b71df1
•
1 Parent(s): 256858e

arg

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ print("Done")
18
  def create_medusa_heads(model_id: str):
19
  parser = distributed_run.get_args_parser()
20
  args = parser.parse_args([
21
- "--nproc_per_node", "4",
22
  "medusa/medusa/train/train.py",
23
  "--model_name_or_path", model_id,
24
  "--data_path", "data/ShareGPT_V4.3_unfiltered_cleaned_split.json",
 
18
  def create_medusa_heads(model_id: str):
19
  parser = distributed_run.get_args_parser()
20
  args = parser.parse_args([
21
+ "--nproc_per_node", "2",
22
  "medusa/medusa/train/train.py",
23
  "--model_name_or_path", model_id,
24
  "--data_path", "data/ShareGPT_V4.3_unfiltered_cleaned_split.json",