Colab Pro notebook from https://github.com/TheLastBen/fast-stable-diffusion Error to initiate the Training

[jhuse25]

Hi Ben,

Firstly, I really appreciate your work and your openness to share with the community your knowledge and tools so we can use them and access to the excitement world of IA, recently I wanted to train a new model with my face but I'm getting and error when I got to the step Training, the Unet is showing an error, however I havent been able to solve it, it is possible that you can help me with that issue please? This is the error I'm getting from Colab:

Thanks in advance!