Browse Source

add img embedding.py

luojunhui 4 weeks ago
parent
commit
217772867e
1 changed files with 3 additions and 3 deletions
  1. 3 3
      applications/clip_embedding/embedding.py

+ 3 - 3
applications/clip_embedding/embedding.py

@@ -34,9 +34,9 @@ async def embed_image_url(img_url_list: List[str]):
 
 
         with torch.no_grad():
         with torch.no_grad():
             # ✅ 调用图像编码器
             # ✅ 调用图像编码器
-            feats = model.get_image_features(pixel_values=pixel_values)
+            image_features = model.encode_image(pixel_values)
 
 
-        feats = _normalize(feats)
-        outputs.extend(_to_list(feats))
+        feats = _normalize(image_features)
+        outputs.extend(_to_list(image_features))
 
 
     return outputs
     return outputs