Update app.py

app.py

@@ -81,7 +81,7 @@ def embed(req: EmbedRequest):
                 texts=[text],
                 task=req.task,
                 prompt_name=req.prompt_name or "query",
-                return_multivector=True,
+                return_multivector=req.return_token_embeddings,
                 truncate_dim=req.truncate_dim,
             )
         # outputs[0] = (num_vectors, hidden_dim)
@@ -107,7 +107,7 @@ def embed(req: EmbedRequest):
                 texts=[tokenizer.decode(window_ids[0])],
                 task=req.task,
                 prompt_name=req.prompt_name or "passage",
-                return_multivector=True,
+                return_multivector=req.return_token_embeddings,
                 truncate_dim=req.truncate_dim,
             )