Skip to content

Finetuning a open source llm (microsoft's phi-2) to perform conversation summarization.

Notifications You must be signed in to change notification settings

Ritaprava95/LLM_Finetuning_for_specialized_task

Repository files navigation

We will finetune an open-source LLM using PEFT, LoRA to perform sepecialized task using Prompt Engineering.

Environment setup

conda create -n llm_finetuning python=3.11
conda install anaconda::jupyter
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
pip install transformers
pip install datasets
pip install peft accelerate
pip install trl
pip install scipy
pip install bitsandbytes

About

Finetuning a open source llm (microsoft's phi-2) to perform conversation summarization.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published