File size: 745 Bytes
db10084 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | """Build-time model prefetch utility for reducing cold-start downloads."""
import asyncio
from app.core.config import settings
from app.core.logging import get_logger, setup_logging
from app.services.model_registry import get_model_registry
setup_logging()
logger = get_logger(__name__)
async def main() -> None:
"""Download fusion and submodel repositories into the configured HF cache."""
logger.info("Starting build-time model prefetch for %s", settings.HF_FUSION_REPO_ID)
registry = get_model_registry()
await registry.load_from_fusion_repo(settings.HF_FUSION_REPO_ID, force_reload=True)
logger.info("Build-time model prefetch completed")
if __name__ == "__main__":
asyncio.run(main())
|