flatten folders and product management

lucidrains · Dec 13, 2022 · d29c40d · d29c40d
1 parent a38af4b
commit d29c40d
Show file tree

Hide file tree

Showing 6 changed files with 4 additions and 1,758 deletions.
diff --git a/README.md b/README.md
@@ -716,6 +716,7 @@ Anything! It is MIT licensed. In other words, you can freely copy / paste for yo
 - [ ] add textual inversion
 - [ ] cleanup self conditioning to be extracted at imagen instantiation
 - [ ] make sure eventual dreambooth works with imagen-video
+- [ ] allow for temporal interpolation at specific stages
 
 ## Citations
 

diff --git a/imagen_pytorch/elucidated_imagen.py b/imagen_pytorch/elucidated_imagen.py
@@ -40,7 +40,7 @@
     unnormalize_zero_to_one,
 )
 
-from imagen_pytorch.imagen_video.imagen_video import (
+from imagen_pytorch.imagen_video import (
     Unet3D,
     resize_video_to
 )

diff --git a/imagen_pytorch/imagen_pytorch.py b/imagen_pytorch/imagen_pytorch.py
@@ -25,7 +25,7 @@
 
 from imagen_pytorch.t5 import t5_encode_text, get_encoded_dim, DEFAULT_T5_NAME
 
-from imagen_pytorch.imagen_video.imagen_video import Unet3D, resize_video_to
+from imagen_pytorch.imagen_video import Unet3D, resize_video_to
 
 # helper functions
 

diff --git a/imagen_pytorch/imagen_video/__init__.py b/imagen_pytorch/imagen_video/__init__.py