Spaces:
Runtime error
Runtime error
yuvalalaluf
commited on
Commit
β’
daafd88
1
Parent(s):
b058ece
Update run.py
Browse files
run.py
CHANGED
@@ -53,11 +53,6 @@ def run_appearance_transfer(model: AppearanceTransferModel, cfg: RunConfig) -> L
|
|
53 |
cross_image_attention_range=Range(start=start_step, end=end_step),
|
54 |
).images
|
55 |
# Save images
|
56 |
-
images[0].save(cfg.output_path / f"out_transfer---seed_{cfg.seed}.png")
|
57 |
-
images[1].save(cfg.output_path / f"out_style---seed_{cfg.seed}.png")
|
58 |
-
images[2].save(cfg.output_path / f"out_struct---seed_{cfg.seed}.png")
|
59 |
-
joined_images = np.concatenate(images[::-1], axis=1)
|
60 |
-
Image.fromarray(joined_images).save(cfg.output_path / f"out_joined---seed_{cfg.seed}.png")
|
61 |
return images
|
62 |
|
63 |
|
|
|
53 |
cross_image_attention_range=Range(start=start_step, end=end_step),
|
54 |
).images
|
55 |
# Save images
|
|
|
|
|
|
|
|
|
|
|
56 |
return images
|
57 |
|
58 |
|