diff --git a/package-lock.json b/package-lock.json
index 1d2840f..0b9152d 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -1,12 +1,12 @@
 {
   "name": "comfyui-api",
-  "version": "1.8.0",
+  "version": "1.8.1",
   "lockfileVersion": 3,
   "requires": true,
   "packages": {
     "": {
       "name": "comfyui-api",
-      "version": "1.8.0",
+      "version": "1.8.1",
       "hasInstallScript": true,
       "license": "MIT",
       "dependencies": {
diff --git a/package.json b/package.json
index 0d7330f..1a846f2 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "comfyui-api",
-  "version": "1.8.0",
+  "version": "1.8.1",
   "description": "Wraps comfyui to make it easier to use as a stateless web service",
   "main": "dist/src/index.js",
   "scripts": {
diff --git a/src/comfy.ts b/src/comfy.ts
index 27798f3..66d1b32 100644
--- a/src/comfy.ts
+++ b/src/comfy.ts
@@ -76,6 +76,7 @@ export async function warmupComfyUI(): Promise<void> {
         "Content-Type": "application/json",
       },
       body: JSON.stringify({ prompt: config.warmupPrompt }),
+      signal: AbortSignal.timeout(1000 * 60 * 60 * 20), // 20 hours
     });
     if (!resp.ok) {
       throw new Error(`Failed to warmup Comfy UI: ${await resp.text()}`);
diff --git a/src/server.ts b/src/server.ts
index 0849666..5f1b293 100644
--- a/src/server.ts
+++ b/src/server.ts
@@ -334,6 +334,7 @@ server.after(() => {
                     filename,
                     prompt,
                   }),
+                  signal: AbortSignal.timeout(1000 * 60 * 60 * 20), // 20 hours
                 })
                   .catch((e: any) => {
                     app.log.error(
@@ -376,6 +377,7 @@ server.after(() => {
                   prompt,
                   error: e.message,
                 }),
+                signal: AbortSignal.timeout(1000 * 60 * 60 * 20), // 20 hours
               });
 
               if (!resp.ok) {
@@ -495,6 +497,7 @@ server.after(() => {
                   "Content-Type": "application/json",
                 },
                 body: JSON.stringify({ prompt, id, webhook, convert_output }),
+                signal: AbortSignal.timeout(1000 * 60 * 60 * 20), // 20 hours
               }
             );
             const body = await resp.json();
diff --git a/test/test-utils.ts b/test/test-utils.ts
index d0018f5..62a9b5e 100644
--- a/test/test-utils.ts
+++ b/test/test-utils.ts
@@ -48,7 +48,7 @@ export async function submitPrompt(
       "Content-Type": "application/json",
     },
     body: JSON.stringify(body),
-    signal: AbortSignal.timeout(1000 * 60 * 60 * 20), // 20 minutes
+    signal: AbortSignal.timeout(1000 * 60 * 60 * 20), // 20 hours
   });
   if (!resp.ok) {
     console.error(await resp.text());