laurenssam commited on
Commit
1886699
·
1 Parent(s): 50a299e
Files changed (1) hide show
  1. app.py +3 -3
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', dtype=torch.float16)
110
 
111
  if type(images) is list:
112
  images = [
113
- image.to(model.device, dtype=torch.float16) for image in images
114
  ]
115
  else:
116
- images = images.to(model.device, dtype=torch.float16)
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):