fix usage of ReadDataFromJson

triton-inference-server · Sep 18, 2024 · a093d02 · a093d02
1 parent 68d4c01
commit a093d02
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/http_server.cc b/src/http_server.cc
@@ -3557,6 +3557,8 @@ HTTPAPIServer::GenerateRequestClass::ExactMappingInput(
       }
     }
 
+	// get original element count back
+    element_cnt = tensor_data.IsArray() ? tensor_data.ArraySize() : 1;
     serialized_data_.emplace_back();
     std::vector<char>& serialized = serialized_data_.back();
     serialized.resize(byte_size);