Spaces:

laurenssam
/

Privacy-Aware-VLM

Sleeping

laurenssam commited on 5 days ago

Commit

2c571f0

1 Parent(s): 1886699

fix dtype

Files changed (1) hide show

app.py CHANGED Viewed

@@ -106,7 +106,7 @@ def get_response(params):
             images = [load_image_from_base64(image) for image in images]
             print(images)
             # images = process_images(images, image_processor, model.config)
-            images = process_images(images, image_processor, model.config).to('cpu')
             if type(images) is list:
                 images = [

             images = [load_image_from_base64(image) for image in images]
             print(images)
             # images = process_images(images, image_processor, model.config)
+            images = process_images(images, image_processor, model.config).to('cpu', dtype=torch.float)
             if type(images) is list:
                 images = [