ValueError: unknown mid_block_type : UNetMidBlock2DCrossAttn

showlab / Tune-A-Video

[ICCV 2023] Tune-A-Video: One-Shot Tuning of Image Diffusion Models for Text-to-Video Generation

https://tuneavideo.github.io

Apache License 2.0

4.15k stars 377 forks source link

ValueError: unknown mid_block_type : UNetMidBlock2DCrossAttn #71

Open piano-accordion opened 1 year ago

piano-accordion commented 1 year ago

when I use my own dreambooth for tune-a-video training, process raise "ValueError: unknown mid_block_type : UNetMidBlock2DCrossAttn", how can I use "UNetMidBlock3DCrossAttn"?

zhangjiewu commented 1 year ago

please refer to https://github.com/showlab/Tune-A-Video/issues/56#issuecomment-1514221869