Art generation with VQGAN + CLIP in Docker, with a simple web UI for anyone with a GPU. A simplified and expanded take on Kevin Costa's work.
aiai-artclipdockerdocker-composegpuhandcodedimagenetpythonpytorchtorchtorchvisionvqganvqgan-clip
1from core.taming.modules.diffusion.attn_block import AttnBlock
2from core.taming.modules.diffusion.resnet_block import ResnetBlock
3
4from core.taming.modules.diffusion.downsample import Downsample
5from core.taming.modules.diffusion.upsample import Upsample
6
7from core.taming.modules.diffusion.encoder import Encoder
8from core.taming.modules.diffusion.decoder import Decoder
9
10
11__all__ = [
12 AttnBlock,
13 ResnetBlock,
14 Downsample,
15 Upsample,
16 Encoder,
17 Decoder,
18]