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)

---

---