[V]

Author: Kye

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "screenai"
-version = "0.0.7"
+version = "0.0.8"
 description = "Screen AI - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <[email protected]>"]

screenai/main.py

@@ -437,10 +437,7 @@ def __init__(
         self.vit_depth = vit_depth
         self.multi_modal_encoder_depth = multi_modal_encoder_depth
         self.llm_decoder_depth = llm_decoder_depth
-
-        patch_height, patch_width = pair(patch_size)
-        channels * patch_height * patch_width
-
+        
         # ViTransformerWrapper
         self.vit = ViTransformerWrapper(
             image_size=image_size,