Skip to content

Commit

Permalink
Merge pull request #74 from BatsResearch/localize-ranked-logits-grpc
Browse files Browse the repository at this point in the history
Update grpc.py
  • Loading branch information
dotpyu authored Jul 5, 2024
2 parents 61b36ef + 52036c1 commit d8a935c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion alfred/fm/remote/grpc.py
Original file line number Diff line number Diff line change
Expand Up @@ -136,7 +136,7 @@ def _run_req_gen():
**{
"prediction": response.message,
"scores": scores,
"logit": logits,
"logits": logits,
"embeddings": bytes_to_tensor(response.embedding),
}
)
Expand Down

0 comments on commit d8a935c

Please sign in to comment.