Upload modeling_mplug_owl2.py with huggingface_hub
Browse files- modeling_mplug_owl2.py +1 -0
modeling_mplug_owl2.py
CHANGED
|
@@ -59,6 +59,7 @@ def tokenizer_image_token(prompt, tokenizer, image_token_index=IMAGE_TOKEN_INDEX
|
|
| 59 |
return input_ids
|
| 60 |
|
| 61 |
def expand2square(pil_img, background_color):
|
|
|
|
| 62 |
width, height = pil_img.size
|
| 63 |
if width == height:
|
| 64 |
return pil_img
|
|
|
|
| 59 |
return input_ids
|
| 60 |
|
| 61 |
def expand2square(pil_img, background_color):
|
| 62 |
+
from PIL import Image
|
| 63 |
width, height = pil_img.size
|
| 64 |
if width == height:
|
| 65 |
return pil_img
|