aljaz/DALLE2-pytorch
1
1
Fork 0
mirror of https://github.com/lucidrains/DALLE2-pytorch.git synced 2025-12-19 01:34:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

refactor blurring training augmentation to be taken care of by the decoder, with option to downsample to previous resolution before upsampling (cascading ddpm). this opens up the possibility of cascading latent ddpm

Browse Source
This commit is contained in:
Phil Wang
2022-04-22 11:09:17 -07:00
parent ad17c69ab6
commit 2c6c91829d
3 changed files with 78 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
dalle2_pytorch/__init__.py
View File

@@ -1,2 +1,4 @@
from dalle2_pytorch.dalle2_pytorch import DALLE2, DiffusionPriorNetwork, DiffusionPrior, Unet, Decoder
from dalle2_pytorch.vqgan_vae import VQGanVAE
from x_clip import CLIP