gg / g4f /client /image_models.py
mfoud444's picture
first commit
79859e3
raw
history blame contribute delete
374 Bytes
from __future__ import annotations
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