Instructions to use Deci/DeciDiffusion-v1-0 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use Deci/DeciDiffusion-v1-0 with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("Deci/DeciDiffusion-v1-0", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- Draw Things
- DiffusionBee
fix pipeline compatibility with diffusers>=0.29
#7
by katuni4ka - opened
- pipeline.py +4 -1
pipeline.py
CHANGED
|
@@ -8,7 +8,10 @@ from diffusers import UNet2DConditionModel
|
|
| 8 |
from diffusers.configuration_utils import register_to_config
|
| 9 |
from diffusers.models.attention import BasicTransformerBlock
|
| 10 |
from diffusers.models.resnet import ResnetBlock2D, Downsample2D, Upsample2D
|
| 11 |
-
|
|
|
|
|
|
|
|
|
|
| 12 |
from diffusers.pipelines.stable_diffusion import StableDiffusionSafetyChecker
|
| 13 |
from diffusers.schedulers import KarrasDiffusionSchedulers
|
| 14 |
from torch import nn
|
|
|
|
| 8 |
from diffusers.configuration_utils import register_to_config
|
| 9 |
from diffusers.models.attention import BasicTransformerBlock
|
| 10 |
from diffusers.models.resnet import ResnetBlock2D, Downsample2D, Upsample2D
|
| 11 |
+
try:
|
| 12 |
+
from diffusers.models.transformer_2d import Transformer2DModelOutput
|
| 13 |
+
except ImportError:
|
| 14 |
+
from diffusers.models.transformers.transformer_2d import Transformer2DModelOutput
|
| 15 |
from diffusers.pipelines.stable_diffusion import StableDiffusionSafetyChecker
|
| 16 |
from diffusers.schedulers import KarrasDiffusionSchedulers
|
| 17 |
from torch import nn
|