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)