Upload tool

blip_tool.py

@@ -47,8 +47,6 @@ class InstructBLIPImageQuestionAnsweringTool(PipelineTool):
             else:
                 self.device = get_default_device()
 
-        # if self.device_map is None:
-        #     self.model.to(self.device)
         self.is_initialized = True
 
     def encode(self, image, question: str):
@@ -57,14 +55,13 @@ class InstructBLIPImageQuestionAnsweringTool(PipelineTool):
     def forward(self, inputs):
         outputs = self.model.generate(
             **inputs,
-            #max_new_tokens=50,
             num_beams=5,
             max_new_tokens=256,
             min_length=1,
             top_p=0.9,
             repetition_penalty=1.5,
             length_penalty=1.0,
-            temperature=1,
+            temperature=0.7,
         )
         return outputs