latent-diffusion
latent-diffusion copied to clipboard
Requirements update + Dockerfile
I was having difficulty running this on a 3090ti due to a pytorch/driver issue. Pytorch kept reporting that the cuda kernels were unavailable, but everything below it was installed correctly. This is usually an unsupported card issue. Pytorch-lightning also need to be updated because of this PR.
I also added a dockerfile and have tested the code in the given container.
Found the other PR with the dockerfile. We can remove the dockerfile from this PR.