components transformers_image_classification_model_import

Image Classification HuggingFace Transformers Model Import

transformers_image_classification_model_import

Overview

Import PyTorch / MLflow model

Version: 0.0.20

View in Studio: https://ml.azure.com/registries/azureml/components/transformers_image_classification_model_import/version/0.0.20

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/labels/latest