Update llava.py

mrG7 · Dec 27, 2023 · 9f96756 · 9f96756
1 parent b0b1c64
commit 9f96756
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vlmeval/vlm/llava.py b/vlmeval/vlm/llava.py
@@ -64,7 +64,7 @@ def build_prompt(self, line, dataset=None):
         question = line['question']
         hint = line['hint'] if ('hint' in line and not pd.isna(line['hint'])) else None
         if hint is not None:
-            question + hint + '\n' + question
+            question = hint + '\n' + question
 
         options = {
             cand: line[cand]
@@ -107,4 +107,4 @@ def generate(self, image_path, prompt, dataset=None):
         with torch.inference_mode():
             output_ids = self.model.generate(input_ids, images=image_tensor, stopping_criteria=[stopping_criteria], **self.kwargs)
         output = self.tokenizer.decode(output_ids[0, input_ids.shape[1]: ]).strip().split("</s>")[0]
-        return output
+        return output