def list_models(return_dict=False, display=False, limit=None):
"""
Returns the list of models that can be used and finetuned with this task.
Args:
return_dict: Default False. True if you want to return in dict form. Otherwise returns list form.
display: Default False. True if you want output printed directly (overrides return_dict, and returns nothing).
limit: Default None. Maximum number of models to return -- leave None to get all models.
"""
return AutoModel.list_models(task=TASK,
return_dict=return_dict,
display=display,
limit=limit,
aliases=LOCAL_ALIASES)
def __init__(self,
model: Union[str, BaseModel] = None,
local: bool = False,
api_key: str = None,
device: str = None):
models = AutoModel.list_models(task=TASK)
super().__init__(model,
local=local,
api_key=api_key,
device=device,
models=models,
task=TASK,
default_local_model=DEFAULT_LOCAL_MODEL)
