修改image output size

ominicontrol.py

@@ -63,6 +63,8 @@ def generate_image(
     }[style]
     pipe.set_adapters(activate_adapter_name)
 
+    original_width, original_height = image.size
+
     factor = 512 / max(image.size)
     image = resize(
         image.resize(
@@ -102,6 +104,12 @@ def generate_image(
         1.5: (960, 640),
     }[ratio]
 
+
+    output_factor = 640 / max(original_width, original_height)
+    width = int(original_width * output_factor)
+    height = int(original_height * output_factor)
+
+
     print(
         f"Image Ratio: {image_ratio}, Inference Mode: {inference_mode}, Image Guidance: {image_guidance}, Seed: {seed}, Steps: {steps}, Size: {width}x{height}"
     )