search

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

cannot import name 'dataclass_transform' from 'typing_extensions' #210

Open LianShuaiLong opened 2 years ago

LianShuaiLong commented 2 years ago

follow the (inference pipeline notebook](https://github.com/borisdayma/dalle-mini/blob/main/tools/inference/inference_pipeline.ipynb) i setup dependencies and libraries: git clone https://github.com/huggingface/transformers.git
cd transformers
python setup.py build
python setup.py install

git clone https://github.com/borisdayma/dalle-mini.git
cd dalle-mini
python setup.py build
python setup.py install

git clone https://github.com/patil-suraj/vqgan-jax.git
cd vqgan-jax
python setup.py build
python setup.py install

when i run the demo the following error: AttributeError: module 'jax' has no attribute 'load_device_count'
ImportError: cannot import name 'dataclass_transform' from 'typing_extensions'

YukiSakuma commented 2 years ago

Try reinstall jax with this !pip install --upgrade jax jaxlib==0.3.7+cuda11.cudnn805 -f https://storage.googleapis.com/jax-releases/jax_releases.html