import os
from huggingface_hub import HfApi, snapshot_download

HF_TOKEN = os.getenv("HF_TOKEN")

api = HfApi()
DEST_REPO = "ibyteohdear/Wan2.2-I2V-14B-Lightning"

print("Fetching Transformer, VAE, and Root Config...")
snapshot_path = snapshot_download("TestOrganizationPleaseIgnore/WAMU_v2_WAN2.2_I2V_LIGHTNING")

print(f"Uploading to {DEST_REPO}...")
api.create_repo(repo_id=DEST_REPO, token=HF_TOKEN, private=False, exist_ok=True)
api.upload_folder(
    folder_path=snapshot_path,
    repo_id=DEST_REPO,
    commit_message="Prepare folders for AOTI compilation script",
    token=HF_TOKEN,
)