wondervictor commited on
Commit
d18eb44
·
verified ·
1 Parent(s): 60383ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -60,9 +60,8 @@ def generate_image_embeddings(prompt_image,
60
  projector,
61
  device='cuda:0'):
62
  prompt_image = prompt_image.convert('RGB')
63
- inputs = vision_processor(images=[prompt_image],
64
- return_tensors="pt",
65
- padding=True)
66
  inputs = inputs.to(device)
67
  image_outputs = vision_encoder(**inputs)
68
  img_feats = image_outputs.image_embeds.view(1, -1)
 
60
  projector,
61
  device='cuda:0'):
62
  prompt_image = prompt_image.convert('RGB')
63
+ inputs = vision_processor(images=prompt_image,
64
+ return_tensors="pt")
 
65
  inputs = inputs.to(device)
66
  image_outputs = vision_encoder(**inputs)
67
  img_feats = image_outputs.image_embeds.view(1, -1)