自定义宽高

2023-03-03 05:01:29 +08:00
parent bc6c50964d
commit a82544b6a9
1 changed files with 2 additions and 0 deletions
--- a/server.py
+++ b/server.py
@@ -310,6 +310,8 @@ def main_dev(opt):

            # 生成图片数
            batch_size = task['number']
+            opt.H = task['height'] if task['height'] else 256
+            opt.W = task['width'] if task['width'] else 256

            if not opt.from_file:
                prompt = task['prompt']