Spaces:
Running
on
Zero
Running
on
Zero
File size: 1,226 Bytes
a602628 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
"""
Download ACE-Step model from HuggingFace
"""
import os
from pathlib import Path
from huggingface_hub import snapshot_download
import logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
def download_model():
"""Download ACE-Step model from HuggingFace."""
model_id = "ACE-Step/ACE-Step-v1-3.5B"
cache_dir = Path("models")
cache_dir.mkdir(exist_ok=True)
logger.info(f"Downloading ACE-Step model: {model_id}")
logger.info("This may take a while (model is ~7GB)...")
try:
model_path = snapshot_download(
repo_id=model_id,
cache_dir=str(cache_dir),
resume_download=True,
local_files_only=False
)
logger.info(f"✅ Model downloaded successfully to: {model_path}")
return model_path
except Exception as e:
logger.error(f"❌ Failed to download model: {e}")
logger.info("\nTroubleshooting:")
logger.info("1. Check your internet connection")
logger.info("2. Verify HuggingFace Hub access")
logger.info("3. Ensure sufficient disk space (~10GB)")
raise
if __name__ == "__main__":
download_model()
|