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
|