Update modeling_GOT.py
Browse files- modeling_GOT.py +2 -0
modeling_GOT.py
CHANGED
|
@@ -504,6 +504,8 @@ class GOTQwenForCausalLM(Qwen2ForCausalLM):
|
|
| 504 |
|
| 505 |
if ocr_type == 'format':
|
| 506 |
qs = 'OCR with format: '
|
|
|
|
|
|
|
| 507 |
else:
|
| 508 |
qs = 'OCR: '
|
| 509 |
|
|
|
|
| 504 |
|
| 505 |
if ocr_type == 'format':
|
| 506 |
qs = 'OCR with format: '
|
| 507 |
+
elif ocr_type == 'VQA':
|
| 508 |
+
qs = 'Answer the following Question :'
|
| 509 |
else:
|
| 510 |
qs = 'OCR: '
|
| 511 |
|