[CLEANUP]

kyegomez · Feb 6, 2024 · 3aeb772 · 3aeb772
1 parent 8f4e79b
commit 3aeb772
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 29 deletions.
diff --git a/Dockerfile b/Dockerfile
diff --git a/lumiere/model.py b/lumiere/model.py
@@ -49,10 +49,6 @@ def __init__(
         self.norm2d = nn.BatchNorm2d(out_channels)
         self.relu = nn.ReLU(inplace=True)
 
-        # Linear projection layer
-        # This will be sized dynamically based on the output of the conv2d layer
-        self.linear = None
-
     def forward(self, x):
         # Input shape: (batch, time, height, width, dimensions)
         # Input shape: (batch, time, height, width, dimensions)

diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "Lumiere"
-version = "0.0.4"
+version = "0.0.5"
 description = "Paper - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <[email protected]>"]

diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,4 @@
 torch
 zetascale
 swarms
+einops