aitextgen
aitextgen copied to clipboard
minimaxir
Small update needed in one Colab notebook
One Colab notebook was updated to specify the pytorch-lightning version but the other one wasn't, so it doesn't work without it.
!pip3 install pytorch-lightning==0.7.6
!pip3 install transformers==2.9.1
!pip3 install fire==0.3.0
- Not working: Finetune OpenAI's 124M GPT-2 model on your own dataset (GPU)
- Working: Train a GPT-2 model + tokenizer from scratch (GPU)
Related to #39 and #41
This one not working https://colab.research.google.com/drive/15qBZx5y9rdaQSyWpsreMDnTiZ5IlN0zD?usp=sharing
I am also facing an error
TypeError: init() got an unexpected keyword argument 'show_progress_bar'
Solution required at urgent basis. Thanks
@NomanSaleem4 did you try adding what I suggested?
!pip3 install pytorch-lightning==0.7.6
!pip3 install transformers==2.9.1
!pip3 install fire==0.3.0
I fixed it by doing that.