23 lines
571 B
Python
23 lines
571 B
Python
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)
|