Fix stray logs from image embeddings (#1931)

This commit is contained in:
Jason Bosco 2024-09-03 20:41:49 -05:00 committed by GitHub
parent 0d030f08fe
commit 05e4671c7d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 4 deletions

View File

@ -30,7 +30,7 @@ embedding_res_t CLIPImageEmbedder::embed(const std::string& encoded_image) {
std::vector<const char*> output_names = {"image_embeds"};
// run inference
LOG(INFO) << "Running image embedder";
// LOG(INFO) << "Running image embedder";
auto output_tensors = session_->Run(Ort::RunOptions{nullptr}, input_names.data(), &input_tensor, 1, output_names.data(), output_names.size());
// get output tensor
@ -108,7 +108,7 @@ std::vector<embedding_res_t> CLIPImageEmbedder::batch_embed(const std::vector<st
// run inference
LOG(INFO) << "Running image embedder";
// LOG(INFO) << "Running image embedder";
auto output_tensors = session_->Run(Ort::RunOptions{nullptr}, input_names.data(), input_tensors.data(), input_tensors.size(), output_names.data(), output_names.size());
// get output tensor

View File

@ -33,7 +33,7 @@ Option<processed_image_t> CLIPImageProcessor::process_image(const std::string& i
// Run inference
std::vector<Ort::Value> output_tensors;
LOG(INFO) << "Running image processor";
// LOG(INFO) << "Running image processor";
try {
output_tensors = session_->Run(Ort::RunOptions{nullptr}, input_names.data(), &input_tensor, 1, output_names.data(), output_names.size());
} catch (...) {
@ -61,7 +61,7 @@ Option<processed_image_t> CLIPImageProcessor::process_image(const std::string& i
}
}
LOG(INFO) << "Image processed";
// LOG(INFO) << "Image processed";
return Option<processed_image_t>(std::move(output));
}