fix strength

flux1_inpaint.py

@@ -24,9 +24,12 @@ def make_inpaint_condition(image, image_mask):
         return None
     
     image[image_mask > 0.5] = -1.0  # set as masked pixel
-    image = np.expand_dims(image, 0).transpose(0, 3, 1, 2)
-    image = torch.from_numpy(image)
-    return image
+
+    image = image.cpu().numpy()  # torch.Tensor を NumPy 配列に変換
+    image = (image * 255).astype(np.uint8)  # 0~1 の範囲を 0~255 に戻し、uint8 型にキャスト
+    image = image.transpose(0, 2, 3, 1)  # (1, 3, H, W) から (1, H, W, 3) に形状を変更
+
+    return image[0]
 
 
 
@@ -40,7 +43,7 @@ def process_image(image,mask_image,prompt="a girl",negative_prompt="",model_id="
         return None
     
     #control_image=make_inpaint_condition(image,mask_image)
-    #image.save("_control.jpg")
+    #mask_image.save("_mask_image.jpg")
     #image = control_image
     
     pipe = FluxInpaintPipeline.from_pretrained(model_id, torch_dtype=torch.bfloat16)
@@ -50,7 +53,7 @@ def process_image(image,mask_image,prompt="a girl",negative_prompt="",model_id="
     generator = torch.Generator(device).manual_seed(seed)
     generators.append(generator)
     
-    output = pipe(prompt=prompt, image=image, mask_image=mask_image,generator=generator)
+    output = pipe(prompt=prompt, image=image, mask_image=mask_image,generator=generator,strength=strength)
 
     return output.images[0]