Stable Diffusion doesn't work after training on dreambooth

[Brandyfan]

Whenever I train up a model through Dreambooth on the colab notebook, I can't test the trained model for some reason.
After training is done and I attempt to test the model, I get this error-

Is there a fix for this?

In the meantime, when I finish training a model, I use the fast stable diffusion notebook to test the model. I'm not sure why stable diffusion won't run using the Dreambooth notebook.

[TheLastBen]

for now, use A1111 notebook for running inference, I will fix that part later

[Brandyfan]

No worries! That's what I've been doing in the meantime.