BenkHel commited on
Commit
d717525
·
verified ·
1 Parent(s): 1170499

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -146,9 +146,10 @@ def generate(state, imagebox, textbox, image_process_mode, temperature, top_p, m
146
  images = process_images(images, image_processor, model.config)
147
 
148
  if type(images) is list:
149
- images = [image.to(model.device, dtype=torch.float16) for image in images]
150
  else:
151
- images = images.to(model.device, dtype=torch.float16)
 
152
 
153
  replace_token = DEFAULT_IMAGE_TOKEN
154
  if getattr(model.config, 'mm_use_im_start_end', False):
 
146
  images = process_images(images, image_processor, model.config)
147
 
148
  if type(images) is list:
149
+ images = [image.to(model.device, dtype=torch.bfloat16) for image in images]
150
  else:
151
+ images = images.to(model.device, dtype=torch.bfloat16)
152
+
153
 
154
  replace_token = DEFAULT_IMAGE_TOKEN
155
  if getattr(model.config, 'mm_use_im_start_end', False):