From 3aeb772a7c267925f9fc9165e1d332e364c38499 Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Tue, 6 Feb 2024 12:28:41 -0800
Subject: [PATCH] [CLEANUP]

---
 Dockerfile       | 24 ------------------------
 lumiere/model.py |  4 ----
 pyproject.toml   |  2 +-
 requirements.txt |  1 +
 4 files changed, 2 insertions(+), 29 deletions(-)
 delete mode 100644 Dockerfile

diff --git a/Dockerfile b/Dockerfile
deleted file mode 100644
index e4d9e8b..0000000
--- a/Dockerfile
+++ /dev/null
@@ -1,24 +0,0 @@
-# ==================================
-# Use an official Python runtime as a parent image
-FROM python:3.10-slim
-RUN apt-get update && apt-get -y install libgl1-mesa-dev libglib2.0-0 build-essential; apt-get clean
-RUN pip install opencv-contrib-python-headless
-
-# Set environment variables
-ENV PYTHONDONTWRITEBYTECODE 1
-ENV PYTHONUNBUFFERED 1
-
-# Set the working directory in the container
-WORKDIR /usr/src/zeta
-
-
-# Install Python dependencies
-# COPY requirements.txt and pyproject.toml if you're using poetry for dependency
-COPY requirements.txt .
-RUN pip install --no-cache-dir --upgrade pip
-RUN pip install --no-cache-dir -r requirements.txt
-
-RUN pip install --no-cache-dir zetascale
-
-# Copy the rest of the application
-COPY . .
diff --git a/lumiere/model.py b/lumiere/model.py
index f2f339d..3bc4a31 100644
--- 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
index c9c6bb1..bb78bc5 100644
--- 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 <kye@apac.ai>"]
diff --git a/requirements.txt b/requirements.txt
index 236a195..9b48337 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,4 @@
 torch
 zetascale
 swarms
+einops