[CLEANUP]

kyegomez · Feb 8, 2024 · 268cdbd · 268cdbd
1 parent e5e11b7
commit 268cdbd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "screenai"
-version = "0.0.5"
+version = "0.0.7"
 description = "Screen AI - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <[email protected]>"]

diff --git a/screenai/main.py b/screenai/main.py
@@ -13,7 +13,6 @@
  Encoder,
  ViTransformerWrapper,
 )
-from einops.layers.torch import Rearrange
 
 # helper functions
 
@@ -439,7 +438,7 @@ def __init__(
 
  image_height, image_width = pair(image_size)
  patch_height, patch_width = pair(patch_size)
- patch_dim = channels * patch_height * patch_width
+ channels * patch_height * patch_width
 
  # ViTransformerWrapper
  self.vit = ViTransformerWrapper(