Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -85,7 +85,7 @@ def embed(req: EmbedRequest):
85
  truncate_dim=req.truncate_dim,
86
  )
87
  # outputs[0] = (num_vectors, hidden_dim)
88
- pooled = outputs[0].mean(dim=0).cpu().tolist()
89
  return {"embeddings": [pooled]}
90
 
91
  # Case 2: Passage → sliding window, token-level embeddings
 
85
  truncate_dim=req.truncate_dim,
86
  )
87
  # outputs[0] = (num_vectors, hidden_dim)
88
+ pooled = outputs[0].mean(dim=0).cpu()
89
  return {"embeddings": [pooled]}
90
 
91
  # Case 2: Passage → sliding window, token-level embeddings