JLNTV_IPD/CogVideo
4
0
Fork 0
mirror of https://github.com/THUDM/CogVideo.git synced 2025-12-02 18:52:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

model_path load for env

Browse Source
This commit is contained in:
glide-the 2024-08-20 15:15:46 +08:00
parent 5287322b70
commit 2e4b1b223b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
inference/gradio_web_demo.py
View File

@ -296,7 +296,7 @@ with gr.Blocks() as demo:
def generate(prompt, num_inference_steps, guidance_scale, progress=gr.Progress(track_tqdm=True)): def generate(prompt, num_inference_steps, guidance_scale, progress=gr.Progress(track_tqdm=True)):
global UP_SCALE_MODEL 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 # Load the upscale model with progress tracking
UP_SCALE_MODEL = load_sd_upscale(UP_SCALE_MODEL_CKPT) UP_SCALE_MODEL = load_sd_upscale(UP_SCALE_MODEL_CKPT)