Skip to content

This project demonstrates how to Fine-Tune Llama-3-8B model on medical data using NVIDIA T4 Tensor Core GPU

Notifications You must be signed in to change notification settings

Siddhesh19991/Llama-3-8B-Fine-tune

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Llama-3-8B-Fine-tune

This repository demonstrates the fine-tuning of the Llama-3-8B model and compares its performance with the base model. It is part of a comparative study between fine-tuning and Retrieval-Augmented Generation (RAG) to determine which approach is more suitable for our use case.

The detailed blog can be found here.

Fine_tuning.ipynb contains all the code necessary for fine-tuning the model.

Dataset

For this project, we will be using publicly available medical data. This dataset is structured as prompt-completion pairs, where users ask medical questions and receive relevant responses from doctors. (Data Source)

You can access the fine-tuned model on Hugging Face via this link.

For questions or feedback about the project, don't hesitate to reach out to me on LinkedIn.

The RAG implementation for this study can be found here.

About

This project demonstrates how to Fine-Tune Llama-3-8B model on medical data using NVIDIA T4 Tensor Core GPU

Topics

Resources

Stars

Watchers

Forks