Sqxww commited on
Commit
e70dd9f
·
1 Parent(s): 47a907f

调整输出size

Browse files
Files changed (1) hide show
  1. ominicontrol.py +3 -1
ominicontrol.py CHANGED
@@ -107,10 +107,12 @@ def generate_image(
107
 
108
 
109
  width, height = image.size
 
 
110
 
111
 
112
  print(
113
- f"Image Ratio: {image_ratio}, Inference Mode: {inference_mode}, Image Guidance: {image_guidance}, Seed: {seed}, Steps: {steps}, Size: {width}x{height}"
114
  )
115
  # Generate
116
  result_img = generate(
 
107
 
108
 
109
  width, height = image.size
110
+ width = int(width * ratio)
111
+ height = int(height * ratio)
112
 
113
 
114
  print(
115
+ f"Image Ratio: {image_ratio}, Inference Mode: {inference_mode}, Image Guidance: {image_guidance}, Seed: {seed}, Steps: {steps}, Ratio: {ratio}, Size: {width}x{height}"
116
  )
117
  # Generate
118
  result_img = generate(