saharmor / dalle-playground

A playground to generate images from any text prompt using Stable Diffusion (past: using DALL-E Mini)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

"No module Named jax" with jax installed

CaptClockobob opened this issue · comments

I'm trying to start the server on WSL Ubuntu 20.04, and I get this error:

--> Starting DALL-E Server. This might take up to two minutes. Traceback (most recent call last): File "app.py", line 18, in <module> from dalle_model import DalleModel File "/mnt/c/Users/mvaml/dallePlayground/dalle-playground/backend/dalle_model.py", line 5, in <module> import jax ModuleNotFoundError: No module named 'jax'

Jax is installed.

Please help.