111
Browse files- 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))
|