fix for the wrong jax version installed

[Marcelovk]

No description provided.

[cdmichaelb]

@Marcelovk this version should be using GPU and CuDNN right? Or is there something else I need to add to make it use GPU? My GPU doesn't appear to be being used.

[Marcelovk]

As far as I understood yes, but you need to have your nvidia drivers running in order and have the nvidia docker container toolkit, as stated in https://github.com/saharmor/dalle-playground/blob/main/README.md?plain=1#L43

Check the result of the command docker-compose run dalle-backend nvidia-smi to see if your driver is being loaded correctly

[cdmichaelb]

docker-compose run dalle-backend nvidia-smi

Looks like it is, I would've thought this would be faster on GPU :P Thanks again for the help.

[Marcelovk]

I have no idea how fast it should be: Here in my Ryzen 7 notebook with a RTX2060 and 16GB of RAM I get one image every 6 seconds on average

[JoshWarn]

How would one go about having fix #79 included? I tried reinstalling/building the docker, libraries, and MEGA as per the instructions in the Dall-e windows installation guide, but was still running into the issue.

[Larry-Schultz]

How would one go about having fix #79 included? I tried reinstalling/building the docker, libraries, and MEGA as per the instructions in the Dall-e windows installation guide, but was still running into the issue.

If you can't switch the branch manually using git, you can try making the exact changes on the files and lines that the commit in this PR makes.

EDIT: remember to do docker-compose build after changing the Dockerfile.

[NoahSchiffman]

I was actually trying to install DALL-E-MEGA on my Windows 11 x64 WSL 2 Ubuntu 22.04 LTS installation and I had run into a bug like this one. Is this that bug?