borisdayma / dalle-mini

DALL·E Mini - Generate images from a text prompt
https://www.craiyon.com
Apache License 2.0
14.75k stars 1.21k forks source link

error installing dalle-mini on macos and on python container #336

Open easybr opened 1 year ago

easybr commented 1 year ago

pip install dalle-mini Collecting dalle-mini Downloading dalle_mini-0.1.4-py3-none-any.whl (34 kB) Collecting transformers==4.25.1 (from dalle-mini) Downloading transformers-4.25.1-py3-none-any.whl (5.8 MB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 5.8/5.8 MB 16.9 MB/s eta 0:00:00 Collecting einops (from dalle-mini) Downloading einops-0.6.1-py3-none-any.whl (42 kB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 42.2/42.2 kB 15.7 MB/s eta 0:00:00 Collecting unidecode (from dalle-mini) Downloading Unidecode-1.3.6-py3-none-any.whl (235 kB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 235.9/235.9 kB 12.7 MB/s eta 0:00:00 Collecting ftfy (from dalle-mini) Downloading ftfy-6.1.1-py3-none-any.whl (53 kB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 53.1/53.1 kB 22.3 MB/s eta 0:00:00 Collecting emoji (from dalle-mini) Downloading emoji-2.7.0.tar.gz (361 kB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 361.8/361.8 kB 17.0 MB/s eta 0:00:00 Installing build dependencies ... done Getting requirements to build wheel ... done Preparing metadata (pyproject.toml) ... done Collecting pillow (from dalle-mini) Downloading Pillow-10.0.0-cp311-cp311-manylinux_2_28_aarch64.whl (3.2 MB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 3.2/3.2 MB 22.1 MB/s eta 0:00:00 Collecting jax==0.3.25 (from dalle-mini) Downloading jax-0.3.25.tar.gz (1.1 MB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 1.1/1.1 MB 21.6 MB/s eta 0:00:00 Preparing metadata (setup.py) ... done Collecting flax==0.6.3 (from dalle-mini) Downloading flax-0.6.3-py3-none-any.whl (197 kB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 197.4/197.4 kB 18.7 MB/s eta 0:00:00 Collecting wandb (from dalle-mini) Downloading wandb-0.15.8-py3-none-any.whl (2.1 MB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 2.1/2.1 MB 22.8 MB/s eta 0:00:00 Collecting numpy>=1.12 (from flax==0.6.3->dalle-mini) Downloading numpy-1.25.2-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (14.2 MB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 14.2/14.2 MB 20.5 MB/s eta 0:00:00 Collecting matplotlib (from flax==0.6.3->dalle-mini) Downloading matplotlib-3.7.2-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (11.4 MB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 11.4/11.4 MB 21.8 MB/s eta 0:00:00 Collecting msgpack (from flax==0.6.3->dalle-mini) Downloading msgpack-1.0.5-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (316 kB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 316.1/316.1 kB 16.1 MB/s eta 0:00:00 Collecting optax (from flax==0.6.3->dalle-mini) Downloading optax-0.1.7-py3-none-any.whl (154 kB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 154.1/154.1 kB 14.5 MB/s eta 0:00:00 Collecting orbax (from flax==0.6.3->dalle-mini) Downloading orbax-0.1.8.tar.gz (1.6 kB) Preparing metadata (setup.py) ... error error: subprocess-exited-with-error

× python setup.py egg_info did not run successfully. │ exit code: 1 ╰─> [3 lines of output]

  *** Orbax is a namespace, and not a standalone package. For model checkpointing and exporting utilities, please install `orbax-checkpoint` and `orbax-export` respectively (instead of `orbax`). ***

  [end of output]

note: This error originates from a subprocess, and is likely not a problem with pip. error: metadata-generation-failed

× Encountered error while generating package metadata. ╰─> See above for output.

note: This is an issue with the package mentioned above, not pip. hint: See above for details.

Vivraan commented 1 year ago

Confirming that this is also happening on Windows 11 22H2 22621.2134.

The solution seems to be run pip install orbax-checkpoint orbax-export dalle-mini.

borisdayma commented 1 year ago

Actually you should run: pip install dalle-mini orbax==0.0.23

borisdayma commented 1 year ago

Now with newest release just pip install --upgrade dalle-mini should work.

Vivraan commented 1 year ago

The Colab notebook is in disrepair.

borisdayma commented 1 year ago

It works fine for me… I have Colab Pro though so can select A100 or V100. I think with free Colab you can only run the smaller version of dalle-mini: dalle-mini/dalle-mini/mini-1:v0

Vivraan commented 1 year ago

It works fine for me… I have Colab Pro though so can select A100 or V100. I think with free Colab you can only run the smaller version of dalle-mini: dalle-mini/dalle-mini/mini-1:v0

I used V100.