import os
import huggingface_hub # type: ignore

def download_models(model_id="thincamel/ootd-models", local_dir="checkpoints"):
    """
    下载模型文件到指定目录
    """
    os.makedirs(local_dir, exist_ok=True)
    
    try:
        huggingface_hub.snapshot_download(
            repo_id=model_id,
            local_dir=local_dir,
            ignore_patterns=["*.md", ".gitattributes"]
        )
        print(f"模型文件已成功下载到 {local_dir}")
    except Exception as e:
        print(f"下载模型时发生错误: {str(e)}")
        raise