Narsil HF staff commited on
Commit
359575a
β€’
1 Parent(s): d3e5ded

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -44,7 +44,7 @@ def create_medusa_heads(model_id: str):
44
  distributed_run.run(args)
45
 
46
  # Upload the medusa heads to the Hub
47
- repo_id = f"medusa-{model_id}"
48
  api = HfApi()
49
  api.create_repo(
50
  repo_id=repo_id,
@@ -87,7 +87,7 @@ def run(model_id: str) -> str:
87
  proc = mp.Process(target=create_medusa_heads, args=(model_id, ))
88
  proc.start()
89
  proc.join()
90
- repo_id = f"medusa-{model_id}"
91
  print("Success βœ…\nMedusa heads uploaded to: ", repo_id)
92
  return f"""
93
  ### Success πŸ”₯
 
44
  distributed_run.run(args)
45
 
46
  # Upload the medusa heads to the Hub
47
+ repo_id = f"{model_id}-medusa"
48
  api = HfApi()
49
  api.create_repo(
50
  repo_id=repo_id,
 
87
  proc = mp.Process(target=create_medusa_heads, args=(model_id, ))
88
  proc.start()
89
  proc.join()
90
+ repo_id = f"{model_id}/medusa"
91
  print("Success βœ…\nMedusa heads uploaded to: ", repo_id)
92
  return f"""
93
  ### Success πŸ”₯