From 910f6f2b2967fa19f37a86521180c0ed659b1603 Mon Sep 17 00:00:00 2001
From: Nader Arbabian <nader@vast.ai>
Date: Tue, 24 Sep 2024 16:09:24 -0700
Subject: [PATCH] fix: fix incorrect check for comfy_model env var

---
 start_server.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/start_server.sh b/start_server.sh
index 12aaa30..45a40ee 100755
--- a/start_server.sh
+++ b/start_server.sh
@@ -26,7 +26,7 @@ function echo_var(){
 [ -z "$BACKEND" ] && echo "BACKEND must be set!" && exit 1
 [ -z "$MODEL_LOG" ] && echo "MODEL_LOG must be set!" && exit 1
 [ -z "$HF_TOKEN" ] && echo "HF_TOKEN must be set!" && exit 1
-[ "$BACKEND" = "sd3" ] && [ -z "$COMFY_MODEL" ] && echo "For sd3 backends, COMFY_MODEL must be set!" && exit 1
+[ "$BACKEND" = "comfyui" ] && [ -z "$COMFY_MODEL" ] && echo "For comfyui backends, COMFY_MODEL must be set!" && exit 1
 
 
 echo "start_server.sh"