Tested on Linux, macOS
- Install the required packages by executing
prepare.sh
- If some packages are not compatible, modify the
requirements.txt
sh ./prepare.sh
- Huggingface API key
- 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.
- wandb API key (optional)
- This is optional but using wandb may help you check the progress of training your model.
Be careful of uploading your credentials!!