diff --git a/server.py b/server.py
index bece1be..c234883 100644
--- 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']