zezeze97 commited on
Commit
ebb28b4
·
1 Parent(s): 42e4305
Files changed (1) hide show
  1. modeling_fegeo_llama.py +1 -0
modeling_fegeo_llama.py CHANGED
@@ -2776,6 +2776,7 @@ class FEGeoLlamaForCausalLM(LlamaForCausalLM, FEGeoMetaForCausalLM):
2776
  image_aspect_ratio = getattr(model_cfg, "image_aspect_ratio", None)
2777
  new_images = []
2778
  if image_aspect_ratio == 'pad':
 
2779
  for image in images:
2780
  image = self.crop(image)
2781
  image = self.expand2square(image, (255, 255, 255))
 
2776
  image_aspect_ratio = getattr(model_cfg, "image_aspect_ratio", None)
2777
  new_images = []
2778
  if image_aspect_ratio == 'pad':
2779
+ print(f'using padding')
2780
  for image in images:
2781
  image = self.crop(image)
2782
  image = self.expand2square(image, (255, 255, 255))