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

now completely OpenAI CLIP compatible for training

Browse Source
This commit is contained in:
Phil Wang
2022-04-29 11:26:24 -07:00
parent f4a54e475e
commit 39d3659ad9
5 changed files with 187 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
dalle2_pytorch/__init__.py
View File

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