from huggingface_hub import snapshot_download
import sys

repo_id = 'Qwen/Qwen3-TTS-12Hz-1.7B-Base'
local_dir = '/opt/ai-avatar-demo/models/qwen3-tts-12hz-1b7-base'
print('DOWNLOAD_BEGIN', repo_id, '->', local_dir)
try:
    path = snapshot_download(
        repo_id=repo_id,
        local_dir=local_dir,
        local_dir_use_symlinks=False,
        resume_download=True
    )
    print('DOWNLOAD_DONE', path)
except Exception as e:
    print('DOWNLOAD_FAILED', repr(e))
    sys.exit(1)