jolibrain / joliGEN

Generative AI Image Toolset with GANs and Diffusion for Real-World Applications
https://www.joligen.com
Other
244 stars 32 forks source link

Feat discriminator unet mha #580

Open wr0124 opened 1 year ago

wr0124 commented 1 year ago

Integrate CutMix into the unet_discriminator_mha which is original UNet.