sayakpaul HF staff commited on
Commit
b0bc114
1 Parent(s): 998c25c

Update unet/my_unet_model.py (#4)

Browse files

- Update unet/my_unet_model.py (c1f4a0eac1247843fb3ea601a2dd43f588fa467f)

Files changed (1) hide show
  1. unet/my_unet_model.py +1 -1
unet/my_unet_model.py CHANGED
@@ -42,7 +42,7 @@ from diffusers.models.embeddings import (
42
  Timesteps,
43
  )
44
  from diffusers.models.modeling_utils import ModelMixin
45
- from diffusers.models.unet_2d_blocks import (
46
  UNetMidBlock2DCrossAttn,
47
  UNetMidBlock2DSimpleCrossAttn,
48
  get_down_block,
 
42
  Timesteps,
43
  )
44
  from diffusers.models.modeling_utils import ModelMixin
45
+ from diffusers.models.unets.unet_2d_blocks import (
46
  UNetMidBlock2DCrossAttn,
47
  UNetMidBlock2DSimpleCrossAttn,
48
  get_down_block,