czczup commited on
Commit
170bf72
1 Parent(s): acf2cf6

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. modeling_internvl_chat.py +0 -2
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)