File size: 571 Bytes
92d8b0d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
from __future__ import annotations

import asyncio

from app.configs.settings import get_settings
from app.services.huggingface_service import HuggingFaceService
from app.services.model_service import ModelService


async def main() -> None:
    settings = get_settings()
    model_service = ModelService(settings)
    huggingface = HuggingFaceService(settings)
    for model in model_service.list_models():
        path = await huggingface.ensure_model_available(model)
        print(f"{model.display_name}: {path}")


if __name__ == "__main__":
    asyncio.run(main())