I wanted to fine-tune the recently released Llama 2. To my surprise there weren’t any ready-to-use packages out there (even for the original llama) that I could just give a JSON file with conversations to train on. It took me forever to get this right so I decided to turn this into a pip package so others don’t have to go through the same pain. Enjoy!