llama-2-finetuning

finetuning llama-2 7b model based on UCI drug recommendation dataset to identify the disease/condition through the feature fields (Drug, rating, reviews)

This project contains two notebooks

1. The finetuning cdde for finetuning llama-2
2. Inferencing using the finetuning code

These notebooks are platform-dependent and the platform used for the project is google colab a 14 GB T4 GPU. The location of the files are specific to drive location mounted to the notebook and the data used in attached to the data folder.