clean up text tokens long

Tariq-Abuhashim · Aug 9, 2023 · a362665 · a362665
1 parent 1afa2a8
commit a362665
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/palme/model.py b/palme/model.py
@@ -117,8 +117,7 @@ def __init__(self):
     def forward(self, text_tokens, images):
         try:
             # if text_tokens.dtype != torch.long:
-            #     text_tokens = text_tokens.long()
-
+            text_tokens = text_tokens.long()
             images = self.vit_model(pixel_values=images)["last_hidden_state"]
             print(images.shape)
             images = self.perceive(images).squeeze(1)

diff --git a/palme/palm.py b/palme/palm.py
@@ -445,6 +445,7 @@ def generate(
 
         return out[..., n:]
 
+
     def forward(
         self,
         x,
-Original file line number
+Diff line change
@@ Expand Up / @@ -445,6 +445,7 @@ def generate( @@
             return out[..., n:]
         def forward(
             self,
             x,
@@ Expand Down @@