new plan: dalle2-pytorch will now include the space time unet, and no…

…t the other way around
xujiangyu · Dec 10, 2022 · b4730fe · b4730fe
1 parent 1e7531c
commit b4730fe
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/README.md b/README.md
@@ -133,7 +133,7 @@ video_as_images_out = unet(video, enable_time = False)
 
 ## Todo
 
-- [ ] wire up <a href="https://github.com/lucidrains/dalle2-pytorch">dalle2-pytorch</a> unet with pseudo 3d convs + spatial temporal attention
+- [ ] make sure dalle2-pytorch can accept `SpaceTimeUnet` for training
 - [ ] give attention the best positional embeddings research has to offer
 - [ ] soup up the attention
 - [ ] offer a function, similar to how MosaicML's approach, that automatically rigs a 2d-unet from dalle2-pytorch to be 3d

diff --git a/setup.py b/setup.py
@@ -18,7 +18,6 @@
     'axial convolutions'
   ],
   install_requires=[
-    'dalle2-pytorch',
     'einops>=0.6',
     'torch>=1.6',
   ],