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

now completely OpenAI CLIP compatible for training

Browse Source 
just take care of the logic for AdamW and transformers

used namedtuples for clip adapter embedding outputs
This commit is contained in:
Phil Wang
2022-04-29 13:05:01 -07:00
parent f4a54e475e
commit 5063d192b6
6 changed files with 221 additions and 24 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