components aoai_finetune_pipeline

AOAI Proxy Finetune Pipeline

aoai_finetune_pipeline

Overview

Pipeline component for proxy fine-tuning with AOAI

Version: 0.0.1

View in Studio: https://ml.azure.com/registries/azureml/components/aoai_finetune_pipeline/version/0.0.1

Inputs

Name	Description	Type	Default	Optional	Enum
endpoint_name	The endpoint name or AOAI resource name.	string		False
endpoint_resource_group	Resource group for the AOAI resource.	string		True
endpoint_subscription	Subscription for the AOAI resource.	string		True
train_dataset	source file/folder for training dataset.	uri_file		False
validation_dataset	source file/folder for validation dataset.	uri_file		True
model	GPT model engine	string	gpt-35-turbo-0613	False	['babbage-002', 'davinci-002', 'gpt-35-turbo-0125', 'gpt-35-turbo-1106', 'gpt-35-turbo-0613', 'gpt-4']
task_type	Dataset type - chat or completion	string		False	['chat', 'completion']
registered_model_name	User-defined registered model name	string		False
n_epochs	Number of training epochs. If set to -1, number of epochs will be determined dynamically based on the input data.	integer	-1	False
batch_size	Global batch size. If set to -1, batch size will be determined dynamically based on the input data.	integer	-1	False
learning_rate_multiplier	The learning rate multiplier to use for training.	number	1.0	False

Outputs

Name	Description	Type
finetune_submit_output	Contains finetune submit output file in JSON/custom class format	uri_file