search

stochasticai / x-stable-diffusion

Real-time inference for Stable Diffusion - 0.88s latency. Covers AITemplate, nvFuser, TensorRT, FlashAttention. Join our Discord communty: https://discord.com/invite/TgHXuSJEk6
https://stochastic.ai
Apache License 2.0
553 stars 35 forks source link

Can't run colab notebook #15

Closed Andreilys closed 2 years ago

Andreilys commented 2 years ago

Was trying to run this colab notebook - https://colab.research.google.com/drive/1m3n2n5bfNpRgWJ8K-xwTvhzrTQETWduq?usp=sharing#scrollTo=ONitKUKN4CCW

and seem to be stuck on this error despite pip installing the libraries in the cell above

"ImportError: Please install the accelerate library to use Diffusers with PyTorch. You can do so by running pip install diffusers[torch]. Or if torch is already installed, you can run pip install accelerate." Screenshot 2022-11-03 at 1 27 16 PM

Toan-Do commented 2 years ago

Hi @Andreilys , thanks for reporting the bug. In the mean time, you can restart the notebook after installing accelerate then rerun the cell 4 (on your image) then it will work. We will fix it soon in our StochasticX library.