I got this error while generating a 64x64 image with 50 steps:
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)
You've run out of VRAM (alloc failed) means it has no more room/space to allocate the needed memory because there is no more, and task fails.
