import os
import sys
from huggingface_hub import snapshot_download

# 设置保存路径
local_dir = os.path.join(os.getcwd(), "DreamOmni2_Model")

print(f"正在下载 DreamOmni2 模型到: {local_dir}")
print("这可能需要一些时间，取决于您的网络速度...")

try:
    snapshot_download(
        repo_id="xiabs/DreamOmni2",
        local_dir=local_dir,
        local_dir_use_symlinks=False,
        resume_download=True,
        max_workers=8
    )
    print("模型下载成功！")
except Exception as e:
    print(f"下载出错: {e}")
    sys.exit(1)