components Foundational Models documentation

Foundational Models

Components in this category

---

• ft_nlp_common_validation

  Component to validate the finetune job against Validation Service

• ft_nlp_model_converter

  Component to convert the finetune job output to pytorch and mlflow model

• image_framework_selector

  Framework selector control flow component for image tasks

• image_model_output_selector

  Model output selector control flow component for image tasks

• mmdetection_image_objectdetection_instancesegmentation_finetune

  Component to finetune MMDetection models for image object detection and instance segmentation.

• mmdetection_image_objectdetection_instancesegmentation_model_import

  Import PyTorch / MLflow model

• mmdetection_image_objectdetection_instancesegmentation_pipeline

  Pipeline component for image object detection and instance segmentation using MMDetection models.

• question_answering_datapreprocess

  Component to preprocess data for question answering task. See docs to learn more.

• question_answering_model_import

  Component to import PyTorch / MLFlow model. See docs to learn more.

• question_answering_pipeline

  Pipeline Component to finetune Hugging Face pretrained models for extractive question answering task. The component supports optimizations such as LoRA, Deepspeed and ONNXRuntime for performance enhancement. See docs to learn more.

• summarization_datapreprocess

  Component to preprocess data for summarization task. See docs to learn more.

• summarization_finetune

  Component to finetune Hugging Face pretrained models for summarization task. The component supports optimizations such as LoRA, Deepspeed and ONNXRuntime for performance enhancement. See docs to learn more.

• summarization_model_import

  Component to import PyTorch / MLFlow model. See docs to learn more.

• summarization_pipeline

  Pipeline Component to finetune Hugging Face pretrained models for summarization task. The component supports optimizations such as LoRA, Deepspeed and ONNXRuntime for performance enhancement. See docs to learn more.

• text_classification_datapreprocess

  Component to preprocess data for single label classification task. See docs to learn more.

• text_classification_finetune

  Component to finetune Hugging Face pretrained models for text classification task. The component supports optimizations such as LoRA, Deepspeed and ONNXRuntime for performance enhancement. See docs to learn more.

• text_classification_model_import

  Component to import PyTorch / MLFlow model. See docs to learn more.

• text_classification_pipeline

  Pipeline component to finetune Hugging Face pretrained models for text classification task. The component supports optimizations such as LoRA, Deepspeed and ONNXRuntime for performance enhancement. See docs to learn more.

• text_generation_datapreprocess

  Component to preprocess data for text generation task

• text_generation_finetune

  Component to finetune model for Text Generation task

• text_generation_model_import

  Import PyTorch / MLFlow model

• text_generation_pipeline

  Pipeline component for text generation

• token_classification_datapreprocess

  Component to preprocess data for token classification task. See docs to learn more.

• token_classification_finetune

  Component to finetune Hugging Face pretrained models for token classification task. The component supports optimizations such as LoRA, Deepspeed and ONNXRuntime for performance enhancement. See docs to learn more.

• token_classification_model_import

  Component to import PyTorch / MLFlow model. See docs to learn more.

• token_classification_pipeline

  Pipeline component to finetune Hugging Face pretrained models for token classification task. The component supports optimizations such as LoRA, Deepspeed and ONNXRuntime for performance enhancement. See docs to learn more.

• transformers_image_classification_finetune

  Component to finetune HuggingFace transformers models for image classification.

• transformers_image_classification_model_import

  Import PyTorch / MLflow model

• transformers_image_classification_pipeline

  Pipeline component for image classification using HuggingFace transformers models.

• translation_datapreprocess

  Component to preprocess data for translation task. See docs to learn more.

• translation_finetune

  Component to finetune Hugging Face pretrained models for translation task. The component supports optimizations such as LoRA, Deepspeed and ONNXRuntime for performance enhancement. See docs to learn more.

• translation_model_import

  Component to import PyTorch / MLFlow model. See docs to learn more.

• translation_pipeline

  Pipeline component to finetune Hugging Face pretrained models for translation task. The component supports optimizations such as LoRA, Deepspeed and ONNXRuntime for performance enhancement. See docs to learn more.