Update modeling_GOT.py

modeling_GOT.py

@@ -570,7 +570,7 @@ class GOTQwenForCausalLM(Qwen2ForCausalLM):
                 do_sample=False,
                 num_beams = 1,
                 no_repeat_ngram_size = 20,
-                # streamer=streamer,
+                streamer=streamer,
                 max_new_tokens=4096,
                 stopping_criteria=[stopping_criteria]
                 )
@@ -656,6 +656,7 @@ class GOTQwenForCausalLM(Qwen2ForCausalLM):
 
                     with smart_open(html_path_2, 'w') as web_f_new:
                         web_f_new.write(new_web)
+        return output_ids
 
     def dynamic_preprocess(self, image, min_num=1, max_num=6, image_size=1024, use_thumbnail=True):
         
@@ -845,3 +846,5 @@ class GOTQwenForCausalLM(Qwen2ForCausalLM):
                 
             with smart_open(html_path_2, 'w') as web_f_new:
                 web_f_new.write(new_web)
+
+        return output_ids