From 2465716169bed6b5354ddda3653ec976bb2e4301 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=A1=9C=E8=8F=AF?= Date: Sat, 18 Feb 2023 22:12:27 +0800 Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E8=AF=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- server.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/server.py b/server.py index 5f6eb14..a6149d4 100644 --- a/server.py +++ b/server.py @@ -298,6 +298,8 @@ def main_dev(opt): start_code = None if opt.fixed_code: start_code = torch.randn([opt.n_samples, opt.C, opt.H // opt.f, opt.W // opt.f], device=device) + + ''' # 切换模型 if opt.torchscript or opt.ipex: transformer = model.cond_stage_model.model @@ -435,8 +437,9 @@ def main_dev(opt): #grid_count += 1 print(f"Your samples are ready and waiting for you here: \n{outpath} \n", f" \nEnjoy.") update_task_status(task=item, status='done', progress=1) # 修改任务状态为完成 + ''' print("任务结束, 等待10s后退出..") - time.sleep(10) + #time.sleep(10) break