andrewdbrown
/

GOT-OCR2_Modified_load_image

Model card Files Files and versions

KpLBaTMaN commited on Feb 7

Commit

e47015d

·

1 Parent(s): 90aefec

code

Files changed (1) hide show

modeling_GOT.py +2 -2

modeling_GOT.py CHANGED Viewed

@@ -599,7 +599,7 @@ class GOTQwenForCausalLM(Qwen2ForCausalLM):
             with autocast_context:
                 output_ids = self.generate(
                     input_ids,
-                    images=[image_tensor_1.unsqueeze(0).half().to(device)],
                     do_sample=False,
                     num_beams=1,
                     no_repeat_ngram_size=20,
@@ -611,7 +611,7 @@ class GOTQwenForCausalLM(Qwen2ForCausalLM):
             with autocast_context:
                 output_ids = self.generate(
                     input_ids,
-                    images=[image_tensor_1.unsqueeze(0).half().to(device)],
                     do_sample=False,
                     num_beams=1,
                     no_repeat_ngram_size=20,

             with autocast_context:
                 output_ids = self.generate(
                     input_ids,
+                    images=[image_tensor_1.unsqueeze(0).to(device)],
                     do_sample=False,
                     num_beams=1,
                     no_repeat_ngram_size=20,
             with autocast_context:
                 output_ids = self.generate(
                     input_ids,
+                    images=[image_tensor_1.unsqueeze(0).to(device)],
                     do_sample=False,
                     num_beams=1,
                     no_repeat_ngram_size=20,