Rob

> > same here as well, tried this code below, but not working. Need other solution. > > !pip install lmdb !pip install torch==2.1.0+cu118 torchvision==0.16.0+cu118 torchaudio==2.1.0 torchtext==0.16.0+cpu torchdata==0.7.0 --index-url https://download.pytorch.org/whl/cu118...

> Check this out [#2615 (comment)](https://github.com/TheLastBen/fast-stable-diffusion/issues/2615#issuecomment-1858931505) Just tried solution #2 in fast-dreambooth on Colab, FWIW, and CUDA still fails as soon as text encoder training begins.