from huggingface_hub import HfApi, HfFolder, upload_folder, create_repo
import os

# 🔐 Authenticate (only needed once)
# You can get your token from https://huggingface.co/settings/tokens

# 📁 Path to your model folder
model_dir = "/work1/deming/shared/medusa_mlp_Llama-3.1-8B_medusa_5_lr_5e-05_layers_1_medusa_mlp_Llama-3.1-8B_medusa_5_lr_5e-05_layers_1"

# 📦 Name of your Hugging Face repo (username/model_name)
repo_id = "seliny2/medusa-llama3-8b"

# ⬆️ Upload the whole folder (includes model, config, tokenizer, etc.)
upload_folder(
    repo_id=repo_id,
    folder_path=model_dir,
    path_in_repo=".",  # Upload everything from model_dir to the root
)

print(f"Model uploaded to https://huggingface.co/{repo_id}")