Skip to content

components text_generation_model_import

github-actions[bot] edited this page Dec 23, 2023 · 47 revisions

Text Generation Model Import

text_generation_model_import

Overview

Import PyTorch / MLFlow model

Version: 0.0.28

View in Studio: https://ml.azure.com/registries/azureml/components/text_generation_model_import/version/0.0.28

Inputs

Model and task name

Name Description Type Default Optional Enum
huggingface_id Input HuggingFace model id. Incase of continual finetuning provide proper id. Models from Hugging Face are subject to third party license terms available on the Hugging Face model details page. It is your responsibility to comply with the model's license terms. string True

Continual-Finetuning model path

Name Description Type Default Optional Enum
pytorch_model_path Input folder path containing pytorch model for further finetuning. Proper model/huggingface id must be passed. custom_model True
mlflow_model_path Input folder path containing mlflow model for further finetuning. Proper model/huggingface id must be passed. mlflow_model True

Output of validation component

Name Description Type Default Optional Enum
validation_output Validation status. uri_file True

Validation parameters

Name Description Type Default Optional Enum
system_properties Validation parameters propagated from pipeline. string True

Outputs

Name Description Type
output_dir folder to store model metadata uri_folder

Environment

azureml://registries/azureml/environments/acft-hf-nlp-gpu/versions/37

Clone this wiki locally