Samoed commited on
Commit
519b6b5
·
verified ·
1 Parent(s): 46f332e

remove `<|begin_of_text|>`

Browse files
Files changed (1) hide show
  1. custom_st.py +1 -1
custom_st.py CHANGED
@@ -67,7 +67,7 @@ class MultiModalTransformer(BaseTransformer):
67
  if sub_item["type"] == "text":
68
  text += sub_item["content"]
69
  elif sub_item["type"] in ["image_bytes", "image_path"]:
70
- text += "<|image|><|begin_of_text|>"
71
  if sub_item["type"] == "image_bytes":
72
  img = Image.open(BytesIO(sub_item["content"])).convert("RGB")
73
  else:
 
67
  if sub_item["type"] == "text":
68
  text += sub_item["content"]
69
  elif sub_item["type"] in ["image_bytes", "image_path"]:
70
+ text += "<|image|>"
71
  if sub_item["type"] == "image_bytes":
72
  img = Image.open(BytesIO(sub_item["content"])).convert("RGB")
73
  else: