diff --git a/alfred/fm/remote/grpc.py b/alfred/fm/remote/grpc.py index 4f4b179..6a7aafe 100644 --- a/alfred/fm/remote/grpc.py +++ b/alfred/fm/remote/grpc.py @@ -136,7 +136,7 @@ def _run_req_gen(): **{ "prediction": response.message, "scores": scores, - "logit": logits, + "logits": logits, "embeddings": bytes_to_tensor(response.embedding), } )