Skip to content

Latest commit

 

History

History
31 lines (19 loc) · 876 Bytes

GUIDE.md

File metadata and controls

31 lines (19 loc) · 876 Bytes

A Step-by-Step Guide

Environment Setup

Tested on Linux, macOS

Install prerequisites

  1. Install the required packages by executing prepare.sh
  2. If some packages are not compatible, modify the requirements.txt
sh ./prepare.sh

Modify .env file

  1. Huggingface API key
    1. Since we are using the huggingface library, you have to set your HF API key. Our Python script will upload the model to HF, so you need to generate WRITE key.
  2. wandb API key (optional)
    1. This is optional but using wandb may help you check the progress of training your model.

Be careful of uploading your credentials!!

Train the model

training_guide.md

Inference

inference_guide.md