This is a Google Colab friendly version of TRL's https://github.com/lvwerra/trl/blob/main/examples/toxicity/scripts/gpt-j-6b-toxicity.py example for ATTIS 2023 with minor modifications.
This is a Google Colab friendly version of TRL's https://github.com/lvwerra/trl/blob/main/examples/toxicity/scripts/gpt-j-6b-toxicity.py example for ATTIS 2023 with minor modifications.