Skip to content

components diffusers_text_to_image_model_import

github-actions[bot] edited this page May 6, 2024 · 9 revisions

Text to Image Diffusers Model Import

diffusers_text_to_image_model_import

Overview

Import PyTorch / MLflow model

Version: 0.0.2

View in Studio: https://ml.azure.com/registries/azureml/components/diffusers_text_to_image_model_import/version/0.0.2

Inputs

Model family

Name Description Type Default Optional Enum
model_family Which framework the model belongs to. string HuggingFaceImage True ['HuggingFaceImage']

Model name

Name Description Type Default Optional Enum
model_name Please select models from AzureML Model Assets for all supported models. For HuggingFace models, which are not supported in AuzreML model registry, input HuggingFace model_name here. The Model will be downloaded from HuggingFace hub using this model_name and are subject to third party license terms available on the HuggingFace model details page. It is the user responsibility to comply with the model's license terms. string True

Continual-Finetuning model path

Name Description Type Default Optional Enum
pytorch_model Pytorch Model registered in AzureML Asset. custom_model True
mlflow_model Mlflow Model registered in AzureML Asset. mlflow_model True
validation_output Validation status. uri_file True
download_from_source Download model directly from HuggingFace instead of system registry boolean False True

Outputs

Name Description Type
output_dir Folder to store model metadata. uri_folder

Environment

azureml://registries/azureml/environments/acft-transformers-image-gpu/versions/29

Clone this wiki locally