Closed americanexplorer13 closed 11 months ago
This does not seem to be a problem with freeu, but rather a mismatch in the number of channels between your backbone features and skip features. The latest diffusers has built-in freeu, you can use it more conveniently. Please refer https://github.com/huggingface/diffusers/blob/main/src/diffusers/models/unet_2d_condition.py line 737 and 761
@lyn-rgb hey, thank you for your answer! Yes, I already find that the code for SimpleCrossAttnUpBlock is not really similar to CrossAttnUpBlock, so yep, I've already fixed it!
I've tried to change your code in order to maintain SimpleCrossAttnUpBlock2D however it seems that shapes doesn't fit up. How can I do it?