hf-internal-testing
/

tiny-sdxl-custom-all

Model card Files Files and versions

Update unet/my_unet_model.py

#4

by sayakpaul HF Staff - opened May 28, 2024

base: refs/heads/main

←

from: refs/pr/4

Discussion Files changed

Files changed (1) hide show

unet/my_unet_model.py +1 -1

unet/my_unet_model.py CHANGED Viewed

@@ -42,7 +42,7 @@ from diffusers.models.embeddings import (
     Timesteps,
 )
 from diffusers.models.modeling_utils import ModelMixin
-from diffusers.models.unet_2d_blocks import (
     UNetMidBlock2DCrossAttn,
     UNetMidBlock2DSimpleCrossAttn,
     get_down_block,

     Timesteps,
 )
 from diffusers.models.modeling_utils import ModelMixin
+from diffusers.models.unets.unet_2d_blocks import (
     UNetMidBlock2DCrossAttn,
     UNetMidBlock2DSimpleCrossAttn,
     get_down_block,