Spaces:
Running
Running
No save needed
Browse files
app.py
CHANGED
@@ -642,8 +642,6 @@ def worker_start_end(input_image, end_image, image_position, prompts, n_prompt,
|
|
642 |
def get_start_latent(input_image, has_end_image, end_image, height, width, vae, gpu, image_encoder, high_vram):
|
643 |
input_image_np = resize_and_center_crop(input_image, target_width=width, target_height=height)
|
644 |
|
645 |
-
Image.fromarray(input_image_np).save(os.path.join(outputs_folder, f'{job_id}_start.png'))
|
646 |
-
|
647 |
input_image_pt = torch.from_numpy(input_image_np).float() / 127.5 - 1
|
648 |
input_image_pt = input_image_pt.permute(2, 0, 1)[None, :, None]
|
649 |
|
@@ -653,8 +651,6 @@ def worker_start_end(input_image, end_image, image_position, prompts, n_prompt,
|
|
653 |
|
654 |
end_image_np = resize_and_center_crop(end_image, target_width=width, target_height=height)
|
655 |
|
656 |
-
Image.fromarray(end_image_np).save(os.path.join(outputs_folder, f'{job_id}_end.png'))
|
657 |
-
|
658 |
end_image_pt = torch.from_numpy(end_image_np).float() / 127.5 - 1
|
659 |
end_image_pt = end_image_pt.permute(2, 0, 1)[None, :, None]
|
660 |
|
|
|
642 |
def get_start_latent(input_image, has_end_image, end_image, height, width, vae, gpu, image_encoder, high_vram):
|
643 |
input_image_np = resize_and_center_crop(input_image, target_width=width, target_height=height)
|
644 |
|
|
|
|
|
645 |
input_image_pt = torch.from_numpy(input_image_np).float() / 127.5 - 1
|
646 |
input_image_pt = input_image_pt.permute(2, 0, 1)[None, :, None]
|
647 |
|
|
|
651 |
|
652 |
end_image_np = resize_and_center_crop(end_image, target_width=width, target_height=height)
|
653 |
|
|
|
|
|
654 |
end_image_pt = torch.from_numpy(end_image_np).float() / 127.5 - 1
|
655 |
end_image_pt = end_image_pt.permute(2, 0, 1)[None, :, None]
|
656 |
|