diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-09-11 20:24:24 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-09-11 20:24:24 +0000 |
| commit | 8fb9c57ed62dcef721d50c1eeb9c20f65c509215 (patch) | |
| tree | e30c32744eba482bf82e488f41d874ddf768eac0 /modules/img2img.py | |
| parent | d97c6f221ff9f97823c7ffce181a243dec895fa1 (diff) | |
| download | stable-diffusion-webui-gfx803-8fb9c57ed62dcef721d50c1eeb9c20f65c509215.tar.gz stable-diffusion-webui-gfx803-8fb9c57ed62dcef721d50c1eeb9c20f65c509215.tar.bz2 stable-diffusion-webui-gfx803-8fb9c57ed62dcef721d50c1eeb9c20f65c509215.zip | |
add half() supporrt for CLIP interrogation
Diffstat (limited to 'modules/img2img.py')
| -rw-r--r-- | modules/img2img.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/modules/img2img.py b/modules/img2img.py index 779f620d..e046b9a9 100644 --- a/modules/img2img.py +++ b/modules/img2img.py @@ -3,6 +3,7 @@ import cv2 import numpy as np
from PIL import Image, ImageOps, ImageChops
+from modules import devices
from modules.processing import Processed, StableDiffusionProcessingImg2Img, process_images
from modules.shared import opts, state
import modules.shared as shared
@@ -131,7 +132,7 @@ def img2img(prompt: str, negative_prompt: str, prompt_style: str, init_img, init upscaler = shared.sd_upscalers[upscaler_index]
img = upscaler.upscale(init_img, init_img.width * 2, init_img.height * 2)
- processing.torch_gc()
+ devices.torch_gc()
grid = images.split_grid(img, tile_w=width, tile_h=height, overlap=upscale_overlap)
|