wondervictor commited on
Commit
36178b5
·
verified ·
1 Parent(s): 0ae5e99

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -71,7 +71,7 @@ def generate_image_embeddings(prompt_image,
71
  img_feats = image_outputs.image_embeds.view(1, -1)
72
  img_feats = img_feats / img_feats.norm(p=2, dim=-1, keepdim=True)
73
  if projector is not None:
74
- img_feats = projector(img_feats)
75
  return img_feats
76
 
77
  # @spaces.GPU
 
71
  img_feats = image_outputs.image_embeds.view(1, -1)
72
  img_feats = img_feats / img_feats.norm(p=2, dim=-1, keepdim=True)
73
  if projector is not None:
74
+ img_feats = projector(img_feats) + img_feats
75
  return img_feats
76
 
77
  # @spaces.GPU