error installing dalle-mini on macos and on python container

[easybr]

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]

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]

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

[borisdayma]

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

[Vivraan]

The Colab notebook is in disrepair.

[borisdayma]

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]

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.