now completely OpenAI CLIP compatible for training

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

View File

@@ -10,7 +10,7 @@ setup(
'dream = dalle2_pytorch.cli:dream'
],
},
version = '0.0.65',
version = '0.0.67',
license='MIT',
description = 'DALL-E 2',
author = 'Phil Wang',
@@ -31,7 +31,7 @@ setup(
'torchvision',
'tqdm',
'vector-quantize-pytorch',
'x-clip>=0.4.4',
'x-clip>=0.5.1',
'youtokentome'
],
classifiers=[