Fix input tensors and regenerate warmup files

gatewayd-io · Mar 15, 2024 · 6ef95ee · 6ef95ee
1 parent 60ee8aa
commit 6ef95ee
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 1 deletion.
diff --git a/sqli_model/1/assets.extra/tf_serving_warmup_requests b/sqli_model/1/assets.extra/tf_serving_warmup_requests
diff --git a/sqli_model/2/assets.extra/tf_serving_warmup_requests b/sqli_model/2/assets.extra/tf_serving_warmup_requests
diff --git a/training/warm_up.py b/training/warm_up.py
@@ -47,7 +47,7 @@
     predict_request = predict_pb2.PredictRequest()
     predict_request.model_spec.name = "sqli_model"
     predict_request.model_spec.signature_name = "serving_default"
-    predict_request.inputs["tokens"].CopyFrom(
+    predict_request.inputs["embedding_input"].CopyFrom(
         tensor_util.make_tensor_proto(query_vec, float32)
     )