change img

app.py

@@ -17,19 +17,23 @@ pipe = pipe.to(device)
 pipe.load_lora_weights(adapter_id)
 pipe.fuse_lora()
 
-def resize(value,img):
+def resize(w,h,img):
     img = Image.open(img)
-    img = img.resize((value,value))
+    img = img.resize((w,h))
     return img
 
 def infer(source_img, prompt, steps, seed, Strength):
     start = time.time()
+    print("开始")
     generator = torch.Generator(device).manual_seed(seed)  
     if int(steps * Strength) < 1:
         steps = math.ceil(1 / max(0.10, Strength))
-    source_image = resize(512, source_img)
+    w, h = source_img.size
+    newW = 512
+    newH = int(h * newW / w)
+    source_image = resize(newW,newH, source_img)
     source_image.save('source.png')
-    image = pipe(prompt, image=source_image, strength=Strength, guidance_scale=0.0, num_inference_steps=steps).images[0]
+    image = pipe(prompt, image=source_image,width=newW,height=newH, strength=Strength, guidance_scale=0.0, num_inference_steps=steps).images[0]
     end = time.time()
     print("步数",steps)
     print("时间",end-start)