From d190b348959cf9ab7331d895155033a6adb89ded Mon Sep 17 00:00:00 2001
From: Kasisnu <5044120+kasisnu@users.noreply.github.com>
Date: Fri, 9 Aug 2024 18:37:57 +0530
Subject: [PATCH] comfy runner: catch requests exceptions appropriately

---
 workflows/comfy/ComfyLauncher/ComfyLivePortraitRunner.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/workflows/comfy/ComfyLauncher/ComfyLivePortraitRunner.py b/workflows/comfy/ComfyLauncher/ComfyLivePortraitRunner.py
index cd64afe..d7fb0b5 100644
--- a/workflows/comfy/ComfyLauncher/ComfyLivePortraitRunner.py
+++ b/workflows/comfy/ComfyLauncher/ComfyLivePortraitRunner.py
@@ -92,7 +92,7 @@ args = parse_args()
 def get_queued_jobs_count(server_address):
     try:
         response = requests.get(f"http://{server_address}/queue", timeout=5)
-    except requests.exceptions.BaseException:
+    except requests.exceptions.RequestException:
         print("Error while checking queue")
         return None
     queue_json = response.json()