Upload folder using huggingface_hub
Browse files
modeling_internvl_chat.py
CHANGED
|
@@ -382,7 +382,6 @@ class InternVLChatModel(PreTrainedModel):
|
|
| 382 |
self.img_context_token_id = img_context_token_id
|
| 383 |
|
| 384 |
from .conversation import get_conv_template
|
| 385 |
-
print(f"template: {self.template}")
|
| 386 |
template = get_conv_template(self.template)
|
| 387 |
image_tokens = IMG_START_TOKEN + IMG_CONTEXT_TOKEN * self.num_image_token + IMG_END_TOKEN
|
| 388 |
template.append_message(template.roles[0], image_tokens + '\n' + question)
|
|
@@ -422,7 +421,6 @@ class InternVLChatModel(PreTrainedModel):
|
|
| 422 |
vit_embeds = visual_features
|
| 423 |
else:
|
| 424 |
vit_embeds = self.extract_feature(pixel_values)
|
| 425 |
-
print(vit_embeds)
|
| 426 |
input_embeds = self.language_model.get_input_embeddings()(input_ids)
|
| 427 |
B, N, C = input_embeds.shape
|
| 428 |
input_embeds = input_embeds.reshape(B * N, C)
|
|
|
|
| 382 |
self.img_context_token_id = img_context_token_id
|
| 383 |
|
| 384 |
from .conversation import get_conv_template
|
|
|
|
| 385 |
template = get_conv_template(self.template)
|
| 386 |
image_tokens = IMG_START_TOKEN + IMG_CONTEXT_TOKEN * self.num_image_token + IMG_END_TOKEN
|
| 387 |
template.append_message(template.roles[0], image_tokens + '\n' + question)
|
|
|
|
| 421 |
vit_embeds = visual_features
|
| 422 |
else:
|
| 423 |
vit_embeds = self.extract_feature(pixel_values)
|
|
|
|
| 424 |
input_embeds = self.language_model.get_input_embeddings()(input_ids)
|
| 425 |
B, N, C = input_embeds.shape
|
| 426 |
input_embeds = input_embeds.reshape(B * N, C)
|