diff options
Diffstat (limited to 'scripts/poor_mans_outpainting.py')
| -rw-r--r-- | scripts/poor_mans_outpainting.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/scripts/poor_mans_outpainting.py b/scripts/poor_mans_outpainting.py index c029c67f..b0469110 100644 --- a/scripts/poor_mans_outpainting.py +++ b/scripts/poor_mans_outpainting.py @@ -4,7 +4,7 @@ import modules.scripts as scripts import gradio as gr
from PIL import Image, ImageDraw
-from modules import images, processing
+from modules import images, processing, devices
from modules.processing import Processed, process_images
from modules.shared import opts, cmd_opts, state
@@ -77,7 +77,7 @@ class Script(scripts.Script): mask.height - down - (mask_blur//2 if down > 0 else 0)
), fill="black")
- processing.torch_gc()
+ devices.torch_gc()
grid = images.split_grid(img, tile_w=p.width, tile_h=p.height, overlap=pixels)
grid_mask = images.split_grid(mask, tile_w=p.width, tile_h=p.height, overlap=pixels)
@@ -139,7 +139,7 @@ class Script(scripts.Script): combined_image = images.combine_grid(grid)
if opts.samples_save:
- images.save_image(combined_image, p.outpath_samples, "", initial_seed, p.prompt, opts.grid_format, info=initial_info)
+ images.save_image(combined_image, p.outpath_samples, "", initial_seed, p.prompt, opts.grid_format, info=initial_info, p=p)
processed = Processed(p, [combined_image], initial_seed, initial_info)
|