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/extras.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/extras.py')
| -rw-r--r-- | modules/extras.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/extras.py b/modules/extras.py index 6aeae6cb..40935f98 100644 --- a/modules/extras.py +++ b/modules/extras.py @@ -1,7 +1,7 @@ import numpy as np
from PIL import Image
-from modules import processing, shared, images
+from modules import processing, shared, images, devices
from modules.shared import opts
import modules.gfpgan_model
from modules.ui import plaintext_to_html
@@ -11,7 +11,7 @@ cached_images = {} def run_extras(image, gfpgan_visibility, codeformer_visibility, codeformer_weight, upscaling_resize, extras_upscaler_1, extras_upscaler_2, extras_upscaler_2_visibility):
- processing.torch_gc()
+ devices.torch_gc()
image = image.convert("RGB")
info = ""
|