Spaces:

kirchik47
/

ocr_task

Sleeping

kirchik47 commited on Sep 29, 2024

Commit

4a69b50

1 Parent(s): dcc8a5e

Update modeling_GOT.py

Files changed (1) hide show

custom_got/modeling_GOT.py CHANGED Viewed

@@ -581,7 +581,7 @@ class GOTQwenForCausalLM(Qwen2ForCausalLM):
             with torch.autocast("cpu", dtype=torch.bfloat16):
                 output_ids = self.generate(
                     input_ids,
-                    images=[image_tensor_1.unsqueeze(0).half().cuda()],
                     do_sample=False,
                     num_beams = 1,
                     no_repeat_ngram_size = 20,

             with torch.autocast("cpu", dtype=torch.bfloat16):
                 output_ids = self.generate(
                     input_ids,
+                    images=[image_tensor_1.unsqueeze(0).half()],
                     do_sample=False,
                     num_beams = 1,
                     no_repeat_ngram_size = 20,