Spaces:
Sleeping
Sleeping
Commit
·
1886699
1
Parent(s):
50a299e
debug
Browse files
app.py
CHANGED
@@ -106,14 +106,14 @@ def get_response(params):
|
|
106 |
images = [load_image_from_base64(image) for image in images]
|
107 |
print(images)
|
108 |
# images = process_images(images, image_processor, model.config)
|
109 |
-
images = process_images(images, image_processor, model.config).to('cpu'
|
110 |
|
111 |
if type(images) is list:
|
112 |
images = [
|
113 |
-
image.to(model.device
|
114 |
]
|
115 |
else:
|
116 |
-
images = images.to(model.device
|
117 |
|
118 |
replace_token = DEFAULT_IMAGE_TOKEN
|
119 |
if getattr(model.config, "mm_use_im_start_end", False):
|
|
|
106 |
images = [load_image_from_base64(image) for image in images]
|
107 |
print(images)
|
108 |
# images = process_images(images, image_processor, model.config)
|
109 |
+
images = process_images(images, image_processor, model.config).to('cpu')
|
110 |
|
111 |
if type(images) is list:
|
112 |
images = [
|
113 |
+
image.to(model.device) for image in images
|
114 |
]
|
115 |
else:
|
116 |
+
images = images.to(model.device)
|
117 |
|
118 |
replace_token = DEFAULT_IMAGE_TOKEN
|
119 |
if getattr(model.config, "mm_use_im_start_end", False):
|