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)
|