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

Added single GPU training script for decoder (#108)

Browse Source 
Added config files for training

Changed example image generation to be more efficient

Added configuration description to README

Removed unused import
This commit is contained in:
Aidan Dempster
2022-05-20 22:46:19 -04:00
committed by GitHub
parent 430961cb97
commit 022c94e443
7 changed files with 801 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
setup.py
View File

@@ -41,7 +41,8 @@ setup(
'x-clip>=0.4.4',
'youtokentome',
'webdataset>=0.2.5',
'fsspec>=2022.1.0'
'fsspec>=2022.1.0',
'torchmetrics[image]>=0.8.0'
],
classifiers=[
'Development Status :: 4 - Beta',