Spaces:

uw-insight-lab
/

Probing-Vis-Literacy-of-VLMs

Paused

AustingDong commited on Mar 10

Commit

f73c268

1 Parent(s): b742b00

Update cam.py

Files changed (1) hide show

demo/cam.py CHANGED Viewed

@@ -336,7 +336,7 @@ class AttentionGuidedCAMLLaVA(AttentionGuidedCAM):
     def generate_cam(self, inputs, tokenizer, temperature, top_p, class_idx=None, visual_pooling_method="CLS", focus="Visual Encoder"):
         # Forward pass
-        outputs_raw = self.model(**inputs)
         self.model.zero_grad()
         print("outputs_raw", outputs_raw)

     def generate_cam(self, inputs, tokenizer, temperature, top_p, class_idx=None, visual_pooling_method="CLS", focus="Visual Encoder"):
         # Forward pass
+        outputs_raw = self.model(**inputs, num_logits_to_keep=1)
         self.model.zero_grad()
         print("outputs_raw", outputs_raw)