clone data

app.py

@@ -9,13 +9,15 @@ from huggingface_hub import HfApi
 # Clone the medusa repo locally
 print("Cloning the medusa repo locally...")
 Repo.clone_from("https://github.com/FasterDecoding/Medusa.git", "medusa")
+print("Cloning the Vicuna data locally...")
+Repo.clone_from("https://huggingface.co/datasets/Aeala/ShareGPT_Vicuna_unfiltered", "data")
 print("Done")
 
 
 def create_medusa_heads(model_id: str):
     training_args = [
         "--model_name_or_path", model_id,
-        "--data_path", "ShareGPT_Vicuna_unfiltered/ShareGPT_V4.3_unfiltered_cleaned_split.json",
+        "--data_path", "data/ShareGPT_Vicuna_unfiltered/ShareGPT_V4.3_unfiltered_cleaned_split.json",
         "--bf16", "True",
         "--output_dir", "medusa_heads",
         "--num_train_epochs", "1",
@@ -86,6 +88,7 @@ def run(model_id: str) -> str:
         Yay! Medusa heads were successfully created and uploaded to, {repo_id}
         """
     except Exception as e:
+        print("Error ❌\n", e)
         return f"""
         ### Error 😢😢😢