search

facebookincubator / AITemplate

AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.
Apache License 2.0
4.56k stars 370 forks source link

Seamless images in stable diffusion #45

Open gexahedron opened 2 years ago

gexahedron commented 2 years ago

Hi! There's a neat hack to create seamless images in stable diffusion - by replacing padding_mode in all conv2d to "circular", for vae and for unet: https://github.com/sd-webui/stable-diffusion-webui/discussions/224#discussioncomment-3602679 Is it possible to do the same with AITemplate?

terrychenism commented 2 years ago

That's super interesting, but unfortunately circular padding is not supported in v0.1 release.

antinucleon commented 2 years ago

@terrychenism maybe we can prioritize it and make a v0.11/v0.12 release.

terrychenism commented 2 years ago

@terrychenism maybe we can prioritize it and make a v0.11/v0.12 release.

Yes add it into wishlist.

teddybearxzh commented 1 year ago

Hi is this issue solved by the commit? Will the compiled AIT model now use customized padding mode?