xXAI-botXx / Calm-down-bot

A GPT2 model is to be extended to a new task via transfer learning. The GPT2 model is to be fine-tuned in such a way that it generates interactive, meditative and calming stories for the user.
Other
0 stars 0 forks source link

Create and fine tune GPT2 Model #2

Closed xXAI-botXx closed 9 months ago

xXAI-botXx commented 10 months ago

Download the GPT2 model, prepare the train data and fine tune the model.

xXAI-botXx commented 9 months ago

Open an extra branch or repository for it, if needed

xXAI-botXx commented 9 months ago

https://pypi.org/project/transformers/