Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|