gen2

kyegomez · Oct 15, 2023 · d5a06b5 · d5a06b5
1 parent 490f918
commit d5a06b5
Show file tree

Hide file tree

Showing 5 changed files with 860 additions and 16 deletions.
diff --git a/README.md b/README.md
@@ -1,9 +1,11 @@
 [![Multi-Modality](agorabanner.png)](https://discord.gg/qUtxnK2NMf)
 
-# Paper-Implementation-Template
-A simple reproducible template to implement AI research papers 
+# Gen2
+Implementation of "Text driven video generation" in pytorch
 
-Paper Link
+This will use the implementation of Gen1 with text as an output
+
+[Paper Link](https://research.runwayml.com/gen2)
 
 # Appreciation
 * Lucidrains
@@ -12,13 +14,7 @@ Paper Link
 
 
 # Install
-
-# Usage
-
-# Architecture
-
-# Todo
-
+``
 
 # License
 MIT

diff --git a/example.py b/example.py
@@ -0,0 +1,10 @@
+
+import torch
+from gen2 import Gen2
+
+model = Gen2()
+
+images = torch.randn(1, 3, 128, 128)
+video = torch.randn(1, 3, 16, 128, 128)
+
+run_out = model.forward(images, video)
diff --git a/gen2/__init__.py b/gen2/__init__.py
@@ -0,0 +1,4 @@
+from gen2.main import Gen2
+
+
+__all__ = ['Gen2']