blob: edaa45922426ede7a2073c7f2091f96cdefc9f3e (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
from __future__ import annotations
from .types import Client, ImageProvider
from ..models import ModelUtils
class ImageModels():
def __init__(self, client):
self.client = client
self.models = ModelUtils.convert
def get(self, name, default=None):
model = self.models.get(name)
if model and model.best_provider:
return model.best_provider
return default
|
