Closed
Description
Describe the bug
I'm running the following command in Google Colab but an error occurs.
Reproduction
!pip install diffusers
import torch
from diffusers import StableCascadeDecoderPipeline, StableCascadePriorPipeline
Logs
ImportError Traceback (most recent call last)
<ipython-input-2-a096c7f68b41> in <cell line: 2>()
1 import torch
----> 2 from diffusers import StableCascadeDecoderPipeline, StableCascadePriorPipeline
3
4 prior = StableCascadePriorPipeline.from_pretrained('stabilityai/stable-cascade-prior', torch_dtype=torch.bfloat16).to('cuda')
5 decoder = StableCascadeDecoderPipeline.from_pretrained('stabilityai/stable-cascade', torch_dtype=torch.float16).to('cuda')
ImportError: cannot import name 'StableCascadeDecoderPipeline' from 'diffusers' (/usr/local/lib/python3.10/dist-packages/diffusers/__init__.py)
---------------------------------------------------------------------------
NOTE: If your import is failing due to a missing package, you can
manually install dependencies using either !pip or !apt.
To view examples of installing some common dependencies, click the
"Open Examples" button below.
---------------------------------------------------------------------------System Info
diffusersversion: 0.26.3- Platform: Linux-6.1.58+-x86_64-with-glibc2.35
- Python version: 3.10.12
- PyTorch version (GPU?): 2.1.0+cu121 (True)
- Huggingface_hub version: 0.20.3
- Transformers version: 4.37.2
- Accelerate version: not installed
- xFormers version: not installed
- Using GPU in script?: Yes
- Using distributed or parallel set-up in script?: Don't know