/Finetune_GPTNEO_GPTJ6B

Repo for fine-tuning GPTJ and other GPT models

Primary LanguagePythonGNU Affero General Public License v3.0AGPL-3.0

Finetune_GPTNEO_GPTJ6B

Overview

This repo contains code to fine-tune GPT-J-6B with a famous quotes dataset. Originally, the repo downloaded and converted the model weights when GPTJ was not yet added to huggingface transformer package. That code can still be seen under the branch original_youtube

/quotes_dataset contains the dataset properly formatted for fine-tuning. See repo for making this dataset here

/finetuning_repo contains code orginally from the repo here that I have modified to work with GPT-J-6B

Walkthrough

See the video for orignal repo code here for a video tutorial

  1. First create a conda envrionment and enter the environment
  2. Run the ./install_requirements.sh script
  3. Then you want to copy the data from train.csv and validation.csv from /quotes_dataset to the /finetuning_repo folder
  4. Run the finetuning code with appropriate flags to fine tune the model. See example_run.txt