Exposing models elementary blocks to design new models

[zaccharieramzi]

Is your feature request related to a problem? Please describe.
I would like to implement a new model following the diffusers API, namely the conditional UNet1D (mentioned here, but saying that it will not be implemented at the moment).
To do so, I would like to reuse some of the elementary 1D blocks, like DownBlock1D or UpBlock1D.
Unfortunately they are not available for import.

Describe the solution you'd like.
I would like to be able to write the following import statement without a ModuleNotFoundError.

from diffusers.models.unet.unet_1d_blocks import (
    DownBlock1D,
    UpBlock1D,
)

Describe alternatives you've considered.
My current mitigation is to simply copy-paste the code.

[yiyixuxu]

Hi @zaccharieramzi
sorry I'm not able to reproduce your issue
are you using latest version of diffusers?

[zaccharieramzi]

Here is a colab reproducing the issue: https://colab.research.google.com/drive/1cYnUueid_musKd__QxHHwcRwqtggeYQx?usp=sharing

[zaccharieramzi]

Sorry there was a typo in the import, it's actually

from diffusers.models.unets.unet_1d_blocks import (
    DownBlock1D,
    UpBlock1D,
)

I was missing an "s" at the end of "unets".

[kkj15dk]

Did you ever make a conditional 1D U-net. I would really like to make one which is compatible with the diffusers library.