components aoai_finetuning

AOAI Finetuning Job

aoai_finetuning

Overview

Upload data to Azure OpenAI resource, finetune model and delete data

Version: 0.0.1

View in Studio: https://ml.azure.com/registries/azureml/components/aoai_finetuning/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
training_file_path	jsonl source file/folder for training dataset.	uri_file		False
validation_file_path	source file/folder for validation dataset.	uri_file		True
model	GPT model engine	string	gpt-35-turbo-0613	False
task_type	Dataset type - chat or completion	string		False	['chat', 'completion']
n_epochs	Number of training epochs. If not provided, number of epochs will be determined dynamically based on the input data.	integer		True
batch_size	Global batch size. If not provided, batch size will be determined dynamically based on the input data.	integer		True
learning_rate_multiplier	The learning rate multiplier to use for training. If not provided, batch size will be determined dynamically based on the input data.	number		True
suffix	A string of up to 18 characters that will be added to your fine-tuned model name	string		True

Outputs

Name	Description	Type
aoai_finetuning_output	Contains finetuned model id in output file in JSON/custom class format	uri_file

Environment

azureml://registries/azureml-staging/environments/aoai-data-upload-finetune/versions/1