rayli commited on
Commit
30b64bc
1 Parent(s): 392bda6

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -16,7 +16,7 @@ from diffusers.configuration_utils import ConfigMixin, register_to_config
16
  from diffusers.loaders import UNet2DConditionLoadersMixin
17
  from diffusers.utils import BaseOutput, logging, is_torch_version
18
  from diffusers.models.modeling_utils import ModelMixin
19
- from diffusers.models.unet_2d_blocks import (
20
  CrossAttnDownBlock2D,
21
  CrossAttnUpBlock2D,
22
  DownBlock2D,
 
16
  from diffusers.loaders import UNet2DConditionLoadersMixin
17
  from diffusers.utils import BaseOutput, logging, is_torch_version
18
  from diffusers.models.modeling_utils import ModelMixin
19
+ from diffusers.models.unets.unet_2d_blocks import (
20
  CrossAttnDownBlock2D,
21
  CrossAttnUpBlock2D,
22
  DownBlock2D,