qresearch
/

llama-3-vision-alpha

Model card Files Files and versions Community

qtnx commited on Apr 26

Commit

9c897db

•

1 Parent(s): a84d3da

Update main.py

Files changed (1) hide show

__main__.py +1 -3

__main__.py CHANGED Viewed

@@ -161,9 +161,7 @@ def answer_question(
         image_features = image_forward_outs.hidden_states[-2]
-        image_features2 = image_features[:, 1:]
-        projected_embeddings = projection_module(image_features2).to("cuda")
         embedding_layer = model.get_input_embeddings()
         # text_embeddings = embedding_layer(input_ids)

         image_features = image_forward_outs.hidden_states[-2]
+        projected_embeddings = projection_module(image_features).to("cuda")
         embedding_layer = model.get_input_embeddings()
         # text_embeddings = embedding_layer(input_ids)