From 2e4b1b223ba3404f1a01ea97c840596dfc1410bc Mon Sep 17 00:00:00 2001
From: glide-the <2533736852@qq.com>
Date: Tue, 20 Aug 2024 15:15:46 +0800
Subject: [PATCH] model_path load for  env

---
 inference/gradio_web_demo.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/inference/gradio_web_demo.py b/inference/gradio_web_demo.py
index 2757011..a9fa6e5 100644
--- a/inference/gradio_web_demo.py
+++ b/inference/gradio_web_demo.py
@@ -296,7 +296,7 @@ with gr.Blocks() as demo:
 
     def generate(prompt, num_inference_steps, guidance_scale, progress=gr.Progress(track_tqdm=True)):
         global UP_SCALE_MODEL
-        if not UP_SCALE_MODEL:
+        if not UP_SCALE_MODEL and len(str(UP_SCALE_MODEL_CKPT).strip()) > 0:
             # Load the upscale model with progress tracking
             UP_SCALE_MODEL = load_sd_upscale(UP_SCALE_MODEL_CKPT)