Import PyTorch / MLflow model

Version: 0.0.9

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

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

Name	Description	Type
output_dir	Folder to store model metadata.	uri_folder

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