no encoder cleanup

kyegomez · Aug 17, 2023 · e6d31a4 · e6d31a4
1 parent 16bd2c7
commit e6d31a4
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/medpalm/model.py b/medpalm/model.py
@@ -162,8 +162,11 @@ def forward(self, img, text_tokens, **kwargs):
         - output from the decoder
         """
         try:
-            encoded_img = self.encoder(img, return_embeddings=True)
-            return self.decoder(text_tokens, context=encoded_img)
+            # encoded_img = self.encoder(img, return_embeddings=True)
+            # return self.decoder(text_tokens, context=encoded_img)
+
+            encoded = self.encoder(img, return_embeddings=True)
+            return self.decoder(text_tokens, context=encoded)
         except Exception as error:
             print(f"Failed in forward method: {error}")
             raise