From 14f83ddb7078ca826130eea67b86504819a77232 Mon Sep 17 00:00:00 2001
From: hiro <vuonghoainam.work@gmail.com>
Date: Tue, 12 Dec 2023 07:27:25 +0700
Subject: [PATCH] fix: Revert drogon thread to 1 instead of CPU num

---
 .../inference-nitro-extension/src/module.ts     | 17 ++++-------------
 1 file changed, 4 insertions(+), 13 deletions(-)

diff --git a/extensions/inference-nitro-extension/src/module.ts b/extensions/inference-nitro-extension/src/module.ts
index 266566e91..047581dbe 100644
--- a/extensions/inference-nitro-extension/src/module.ts
+++ b/extensions/inference-nitro-extension/src/module.ts
@@ -179,10 +179,8 @@ async function spawnNitroProcess(nitroResourceProbe: any): Promise<any> {
     let binaryName;
 
     if (process.platform === "win32") {
-      // Todo: Need to check for CUDA support to switch between CUDA and non-CUDA binaries
       binaryName = "win-start.bat";
     } else if (process.platform === "darwin") {
-      // Mac OS platform
       if (process.arch === "arm64") {
         binaryFolder = path.join(binaryFolder, "mac-arm64");
       } else {
@@ -190,21 +188,15 @@ async function spawnNitroProcess(nitroResourceProbe: any): Promise<any> {
       }
       binaryName = "nitro";
     } else {
-      // Linux
-      // Todo: Need to check for CUDA support to switch between CUDA and non-CUDA binaries
-      binaryName = "linux-start.sh"; // For other platforms
+      binaryName = "linux-start.sh";
     }
 
     const binaryPath = path.join(binaryFolder, binaryName);
 
     // Execute the binary
-    subprocess = spawn(
-      binaryPath,
-      [nitroResourceProbe.numCpuPhysicalCore, "127.0.0.1", PORT],
-      {
-        cwd: binaryFolder,
-      }
-    );
+    subprocess = spawn(binaryPath, [1, LOCAL_HOST, PORT], {
+      cwd: binaryFolder,
+    });
 
     // Handle subprocess output
     subprocess.stdout.on("data", (data) => {
@@ -274,7 +266,6 @@ function validateModelVersion(): Promise<void> {
   });
 }
 
-
 function dispose() {
   // clean other registered resources here
   killSubprocess();