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

support CoCa, which seems to be better than CLIP (has an autoregressive text encoder) https://arxiv.org/abs/2205.01917

Browse Source
This commit is contained in:
Phil Wang
2022-05-06 08:27:12 -07:00
parent 98df1ba51e
commit 0be1e0d64c
3 changed files with 57 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
setup.py
View File

@@ -10,7 +10,7 @@ setup(
'dream = dalle2_pytorch.cli:dream'
],
},
version = '0.0.108',
version = '0.0.109',
license='MIT',
description = 'DALL-E 2',
author = 'Phil Wang',
@@ -24,6 +24,7 @@ setup(
install_requires=[
'click',
'clip-anytorch',
'coca-pytorch>=0.0.5',
'einops>=0.4',
'einops-exts>=0.0.3',
'embedding-reader',