repos
/ ai-art master

ai-art

mirror archived upstream

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

337 B · 14 lines · Python Raw History
 1from core.utils.make_cutouts import MakeCutouts
 2from core.utils.normalize import Normalize
 3from core.utils.helpers import resize_image, get_optimizer, get_scheduler, load_vqgan_model, global_seed
 4
 5__all__ = [
 6    MakeCutouts,
 7    Normalize,
 8    resize_image,
 9    get_optimizer,
10    get_scheduler,
11    load_vqgan_model,
12    global_seed
13]