damerajee commited on
Commit
1ff614d
·
verified ·
1 Parent(s): 409b074

Update modeling_gpt2vision.py

Browse files
Files changed (1) hide show
  1. modeling_gpt2vision.py +2 -0
modeling_gpt2vision.py CHANGED
@@ -72,7 +72,9 @@ class GPT2Vision(PreTrainedModel):
72
  }
73
 
74
  with torch.no_grad():
 
75
  inputs_embeds = self.input_embeds(prompt, image_embeds, tokenizer)
 
76
  output_ids = self.text_model.generate(
77
  inputs_embeds=inputs_embeds, **generate_config
78
  )
 
72
  }
73
 
74
  with torch.no_grad():
75
+
76
  inputs_embeds = self.input_embeds(prompt, image_embeds, tokenizer)
77
+ print("inputs_embeds",inputs_embeds.size())
78
  output_ids = self.text_model.generate(
79
  inputs_embeds=inputs_embeds, **generate_config
80
  )