From 8ab70b5aa0ec991e1c4ba39e49166bd5353e2dda Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=A1=9C=E8=8F=AF?= <huan0016@gmail.com>
Date: Wed, 1 Mar 2023 01:58:28 +0800
Subject: [PATCH] =?UTF-8?q?=E9=9A=8F=E6=9C=BA=E6=95=B0=E7=A7=8D=E5=AD=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 server.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/server.py b/server.py
index c16e773..7c5ef4e 100644
--- a/server.py
+++ b/server.py
@@ -280,7 +280,7 @@ def main_dev(opt):
             wm_encoder = WatermarkEncoder()
             wm_encoder.set_watermark('bytes', wm.encode('utf-8'))
 
-            # x
+            # 生成图片数
             batch_size = task['number']
 
             if not opt.from_file:
@@ -303,8 +303,8 @@ def main_dev(opt):
 
             # x
             start_code = None
-            if opt.fixed_code:
-                start_code = torch.randn([task['number'], opt.C, opt.H // opt.f, opt.W // opt.f], device=device)
+            if task['seed']: #if opt.fixed_code:
+                start_code = torch.randn([task['seed'], opt.C, opt.H // opt.f, opt.W // opt.f], device=device)
 
             # 更新进度
             update_task_status(task, "running", 0.8)