Transformers schemas
Predefined inputs and outputs of transformers predictors
Module: implementation.predictors
TransformersBasicInput
Subclass of IOModel.
__init__
Arguments:
inputs (Any)
TransformersBasicOutput
Subclass of IOModel.
__init__
Arguments:
output (Any)
TransformersLogitsOutput
Subclass of IOModel.
__ init__
Arguments:
logits (Any)
TransformersImageClassificationModelInput
Subclass of IOModel.
__init__
Arguments:
pixel_values (Any): Image representation.
TransformersTextToSpeechInput
Subclass of IOModel.
__init__
Arguments:
text_inputs (str): Text to process.
TransformersTextToSpeechOutput
Subclass of IOModel.
__init__
Arguments:
audio (Any): Audio data.
sampling_rate (int): Samling rate.
TransformersChartsAndPlotsModelInput
Subclass of IOModel.
__init__
Arguments:
flattened_patches (Any)
attention_mask (Any)
TransformersVisualQandAInput
Subclass of IOModel.
__init__
Arguments:
image (Image.Image): Input image.
question (str)
TransformersImageModelInput
Subclass of IOModel.
__init__
Arguments:
input_ids (Any)
token_type_ids (Any)
attention_mask (Any)
pixel_values (Any)
pixel_mask (Any)
TransformersEmbeddingInput
Subclass of IOModel.
__init__
Arguments:
encodings (Any)
TransformersEmbeddingOutput
Subclass of IOModel.
__init__
Arguments:
last_hidden_state (Any)
TransformersEntityLinkingInput
Subclass of IOModel.
__init__
Arguments:
encodings (Any)
num_beams (int)
num_return_sequences (int)
prefix_allowed_tokens_fn (Callable[[torch.Tensor, int], List[int]])
TransformersEntityLinkingOutput
Subclass of IOModel.
__init__
Arguments:
sequences (Any)
sequences_scores (Optional[Any], optional): Defaults to None.
TransformersTextualQandAInput
Subclass of IOModel.
__init__
Arguments:
question (str)
context (str)
TransformersTextualQandAOutput
Subclass of IOModel.
__init__
Arguments:
answer (Optional[str], optional): Defaults to None.
score (float): Defaults to 0.
TransformersDETROutput
Subclass of IOModel.
__init__
Arguments:
pred_boxes (Any)
logits (Any)
Last updated