damerajee commited on
Commit
bfd6014
·
verified ·
1 Parent(s): 760a46d

Update modeling_gpt2vision.py

Browse files
Files changed (1) hide show
  1. modeling_gpt2vision.py +1 -1
modeling_gpt2vision.py CHANGED
@@ -62,7 +62,7 @@ class GPT2Vision(PreTrainedModel):
62
  input_texts,
63
  padding='max_length',
64
  truncation=True,
65
- max_length=768,
66
  return_tensors="pt",
67
  ).to(device)
68
  pixel_values = self.vision_encoder(images, device)
 
62
  input_texts,
63
  padding='max_length',
64
  truncation=True,
65
+ max_length=384,
66
  return_tensors="pt",
67
  ).to(device)
68
  pixel_values = self.vision_encoder(images, device)