khang119966
commited on
Commit
•
e829a1d
1
Parent(s):
5ce4270
Update processing_colinternvl2.py
Browse files
processing_colinternvl2.py
CHANGED
@@ -42,13 +42,9 @@ class ColInternVL2Processor(BaseVisualRetrieverProcessor, ProcessorMixin):
|
|
42 |
self.IMG_START_TOKEN='<img>'
|
43 |
self.IMG_END_TOKEN='</img>'
|
44 |
self.img_context_token_id = self.tokenizer.convert_tokens_to_ids(self.IMG_CONTEXT_TOKEN)
|
45 |
-
# self.system_message = '你是由上海人工智能实验室联合商汤科技开发的书生多模态大模型,英文名叫InternVL, 是一个有用无害的人工智能助手。'
|
46 |
self.system_message = 'Bạn là một mô hình trí tuệ nhân tạo đa phương thức Tiếng Việt có tên gọi là Vintern, được phát triển bởi người Việt. Bạn là một trợ lý trí tuệ nhân tạo hữu ích và không gây hại.'
|
47 |
super().__init__(tokenizer)
|
48 |
|
49 |
-
# def from_pretrained(pretrained_model_name_or_path, template="Hermes-2", **kwargs):
|
50 |
-
# return ColInternVL2Processor(pretrained_model_name_or_path, template=template, **kwargs)
|
51 |
-
|
52 |
def build_transform(self, input_size):
|
53 |
MEAN, STD = self.IMAGENET_MEAN, self.IMAGENET_STD
|
54 |
transform = T.Compose([
|
|
|
42 |
self.IMG_START_TOKEN='<img>'
|
43 |
self.IMG_END_TOKEN='</img>'
|
44 |
self.img_context_token_id = self.tokenizer.convert_tokens_to_ids(self.IMG_CONTEXT_TOKEN)
|
|
|
45 |
self.system_message = 'Bạn là một mô hình trí tuệ nhân tạo đa phương thức Tiếng Việt có tên gọi là Vintern, được phát triển bởi người Việt. Bạn là một trợ lý trí tuệ nhân tạo hữu ích và không gây hại.'
|
46 |
super().__init__(tokenizer)
|
47 |
|
|
|
|
|
|
|
48 |
def build_transform(self, input_size):
|
49 |
MEAN, STD = self.IMAGENET_MEAN, self.IMAGENET_STD
|
50 |
transform = T.Compose([
|