chore(trtllm): minor fix

huggingface · Oct 21, 2024 · 42ed868 · 42ed868
1 parent 4a0f05e
commit 42ed868
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/backends/trtllm/include/backend.h b/backends/trtllm/include/backend.h
@@ -44,7 +44,7 @@ namespace huggingface::tgi::backends {
     /**
      * Initialize logging mechanism
      */
-    void huggingface::tgi::backends::InitializeLogging() {
+    void InitializeLogging() {
 #ifdef NDEBUG
         if (const auto TRTLLM_LOG_LEVEL_CSTR = std::getenv("TRTLLM_LOG_LEVEL")) {
         std::string log_level(TRTLLM_LOG_LEVEL_CSTR);

diff --git a/backends/trtllm/lib/backend.cpp b/backends/trtllm/lib/backend.cpp
@@ -24,7 +24,8 @@ void huggingface::tgi::backends::InitializeBackend() {
     }
 }
 
-[[nodiscard]] tle::ParallelConfig GetParallelConfig(const size_t worldSize, std::string workerPath) {
+[[nodiscard]]
+tle::ParallelConfig GetParallelConfig(const size_t worldSize, std::string workerPath) {
     auto mode = tle::CommunicationMode::kLEADER;
     std::optional<tle::OrchestratorConfig> orchestratorConfig = std::nullopt;