From 8392e85891a9f502a41396e81f341ac7b83e7ac1 Mon Sep 17 00:00:00 2001 From: ozanarmagan Date: Tue, 17 Oct 2023 14:31:39 +0300 Subject: [PATCH] Use 'model size * 1.15` --- src/text_embedder_manager.cpp | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/src/text_embedder_manager.cpp b/src/text_embedder_manager.cpp index 912336dc..baa99ebf 100644 --- a/src/text_embedder_manager.cpp +++ b/src/text_embedder_manager.cpp @@ -112,13 +112,12 @@ Option TextEmbedderManager::validate_and_init_local_model(const nlohmann:: const auto& free_memory = SystemMetrics::get_memory_free_bytes(); const auto& model_file_size = std::filesystem::file_size(abs_path); - // return error if model file size is greater than free memory - if(model_file_size > free_memory) { - LOG(ERROR) << "Model file size is greater than free memory: " << model_file_size << " > " << free_memory; - return Option(400, "Model file size is greater than free memory"); + // return error if (model file size * 1.15) is greater than free memory + if(model_file_size * 1.15 > free_memory) { + LOG(ERROR) << "Memory required to load the model exceeds free memory available."; + return Option(400, "Memory required to load the model exceeds free memory available."); } - const std::shared_ptr& embedder = std::make_shared(model_name_without_namespace); auto validate_op = embedder->validate();