Update modeling_GOT.py
Browse files- modeling_GOT.py +3 -3
modeling_GOT.py
CHANGED
@@ -12,7 +12,7 @@ from .got_vision_b import build_GOT_vit_b
|
|
12 |
from torchvision import transforms
|
13 |
from torchvision.transforms.functional import InterpolationMode
|
14 |
import dataclasses
|
15 |
-
|
16 |
|
17 |
DEFAULT_IMAGE_TOKEN = "<image>"
|
18 |
DEFAULT_IMAGE_PATCH_TOKEN = '<imgpad>'
|
@@ -666,7 +666,7 @@ class GOTQwenForCausalLM(Qwen2ForCausalLM):
|
|
666 |
lines = lines.split("const text =")
|
667 |
new_web = lines[0] + gt + lines[1]
|
668 |
|
669 |
-
with
|
670 |
web_f_new.write(new_web)
|
671 |
return response_str
|
672 |
|
@@ -875,7 +875,7 @@ class GOTQwenForCausalLM(Qwen2ForCausalLM):
|
|
875 |
lines = lines.split("const text =")
|
876 |
new_web = lines[0] + 'const text =' + gt + lines[1]
|
877 |
|
878 |
-
with
|
879 |
web_f_new.write(new_web)
|
880 |
|
881 |
return response_str
|
|
|
12 |
from torchvision import transforms
|
13 |
from torchvision.transforms.functional import InterpolationMode
|
14 |
import dataclasses
|
15 |
+
|
16 |
|
17 |
DEFAULT_IMAGE_TOKEN = "<image>"
|
18 |
DEFAULT_IMAGE_PATCH_TOKEN = '<imgpad>'
|
|
|
666 |
lines = lines.split("const text =")
|
667 |
new_web = lines[0] + gt + lines[1]
|
668 |
|
669 |
+
with open(html_path_2, 'w') as web_f_new:
|
670 |
web_f_new.write(new_web)
|
671 |
return response_str
|
672 |
|
|
|
875 |
lines = lines.split("const text =")
|
876 |
new_web = lines[0] + 'const text =' + gt + lines[1]
|
877 |
|
878 |
+
with open(html_path_2, 'w') as web_f_new:
|
879 |
web_f_new.write(new_web)
|
880 |
|
881 |
return response_str
|