随机数种子

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)