max area contorl

app.py

@@ -89,8 +89,8 @@ def infer(
     
 
     # 规范宽高：不超过 MAX_IMAGE_SIZE 且对齐到 16
-    width = max(16, min(width, MAX_IMAGE_SIZE)) // 16 * 16
-    height = max(16, min(height, MAX_IMAGE_SIZE)) // 16 * 16
+    width = max(16, min(width, max_area)) // 16 * 16
+    height = max(16, min(height, max_area)) // 16 * 16
 
     # 随机种子
     if randomize_seed:
@@ -98,7 +98,7 @@ def infer(
     generator = torch.Generator(device=device).manual_seed(int(seed))
 
     # 参考图按桶缩放
-    raw_images = resize_by_bucket(refs, resolution=MAX_IMAGE_SIZE)
+    raw_images = resize_by_bucket(refs, resolution=max_area)
 
     # 推理
     with torch.no_grad():