|
@@ -34,9 +34,9 @@ async def embed_image_url(img_url_list: List[str]):
|
|
|
|
|
|
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
|