Fine-tuning

Overview

LLMs in general are typically trained for generic use cases. Fine-tuning LLMs allows users to further train the model to provide more accurate answers for a specific domain or use case. This process involves providing the model with a dataset containing examples of input and output from a specific domain. LLM Labs helps simplify this process by providing a user-friendly way to fine-tune and deploy open-source models, allowing you to tailor LLMs to your exact needs.

Create fine-tuned models

This section guides you through the process of fine-tuning your models in LLM Labs.

Step 1: Set up model

1. Navigate to the Models page.

2. Open My models tab, and then open the Fine-tuned LLMs section. Click on the Create fine-tuned model button.

   
3. Once the creation wizard is opened, you need to set up your project.

   
4. Name your fine-tuned model.

   1. Select a base model that you want to fine-tune. You can select either:

      1. Pre-trained LLMs. Currently, we support:

         1. Amazon Nova Micro

         2. Amazon Nova Lite

         3. Amazon Nova Pro

         4. Amazon Titan Text G1 - Express

         5. Amazon Titan Text G1 - Lite

         6. Cohere Command

         7. Cohere Command Light

         8. Meta Llama 3.1 8B

         9. Meta Llama 3.1 70B

      2. Existing fine-tuned models

5. Choose a dataset. You can upload either a .csv consisting of 2 columns: prompts, expected completions, or you can choose an existing dataset from the library. For the validation dataset, you have 3 options:

   1. Split from selected dataset: Datasaur will split the uploaded dataset and use it for validation data. You will need to configure the validation size using a percentage.

      
   2. Use new dataset: You will need to add a new dataset to use as validation.

      
   3. None: Choose this option if you don't want to add a validation dataset.

      1. Please note that if you select Cohere Command or Cohere Command Light as the base model, you are required to have validation data.

Step 2: Adjust hyperparameters

Next, you will need to configure and adjust the hyperparameters for your fine-tuning project.

The fundamental hyperparameters are:

In addition to the fundamental hyperparameters, there are advanced hyperparameters with preset default values. These hyperparameters are always applied, but you can adjust them for further hyperparameter fine-tuning if desired. The advanced hyperparameters include:

Step 3: Review job

1. The last step of the creation wizard is to review your fine-tuning job before you start the process. In this step, you will need to review all the configurations that you have chosen. The important part of this page is to review the cost that it will take to start the fine-tuning process.

   

   You can also view the predicted cost by clicking the View total predicted cost button on the Costs section.

   

   It will calculate and show you the total predicted cost for starting the fine-tuning process.

   

1. Once you have reviewed the configuration, you will need to check the acknowledgement checkbox.

2. Lastly, click the Start fine-tuning job button to start the fine-tuning process.

3. The training process has now started.

   

1. Once the training process is complete, your model will be available in the My models section.

   

Model management

Model status

There are 7 possible statuses for the fine-tuned model.

1. Training: The model is currently being trained on your dataset. This status indicates that the training process is in progress, and the model is learning from your data.

   
2. Training failed: The model training process has failed due to an error. This status indicates that the training process was interrupted, and you may need to investigate and resolve the issue.

   
3. Stopping training: The model training process is being stopped. This status indicates that someone has chosen to stop the training.

   
4. Training stopped: The model training process has been stopped. This status indicates that the training process has been successfully stopped, and you can’t continue the training once it stopped.

   
5. Not deployed: The model has been trained but has not yet been deployed for use. You can deploy the model to use it in Sandbox.

   
6. Deploying: The model is being deployed for use. This status indicates that the deployment process is in progress, and the model will soon be available for use in Sandbox.

   
7. Deployed: The model has been successfully deployed. This status indicates that the model is now available for use in Sandbox, and you can start using it to generate predictions or responses.

   

Learn more about deploying or undeploying models.

View model details

To view the model details, click the three-dots button in the top right corner of the model card.

Choose the View details menu, and the details of the fine-tuned model will be shown.

Here you can view the dataset, validation dataset, models used, hyperparameter configurations, the creator, and storage cost information.

Delete fine-tune models

To delete the fine-tune model, click the three-dots button in the top right corner of the model card.

Choose the Delete option.

Acknowledge that you understand the impact of deleting the fine-tuned model by checking the acknowledgement checkbox.

After that, confirm the delete process by clicking the Delete model button, and your model will be deleted.

Deploy and undeploy model

To deploy a fine-tuned model:

1. Navigate to the My models tab, and expand the Fine-tuned LLMs section.

2. Click the Deploy model button to start the deployment.

   
3. Once you click the Deploy model button, you can specify the auto undeploy schedule based on your needs.

   
4. Click the Deploy model button to start the deployment process.

   
5. Once deployment is complete, your model will be available to use for experiment in Sandbox. Learn more about Sandbox.

   

To undeploy a model:

1. Navigate to the My models tab, and expand the Fine-tuned LLMs section.

2. Click the three-dots button in the right corner of the model card.

   
3. Choose the Undeploy option.

   
4. Confirm the undeployment by clicking the Undeploy button, and your model will be undeployed. You will no longer be charged for the hourly cost.

Access via Sandbox

Once the fine-tuned model is deployed, it will be available in the Sandbox for further experimentation and testing. This allows you to integrate and test the specialized model within your specific applications. Learn more about Sandbox.