aboutsummaryrefslogtreecommitdiffstats
path: root/modules/extras.py
diff options
context:
space:
mode:
authorEyeDeck <eyedeck@gmail.com>2022-09-11 22:15:30 +0000
committerGitHub <noreply@github.com>2022-09-11 22:15:30 +0000
commite05e46aa3f3b2230affe399372ad96d8736e4010 (patch)
treee51f423443ad2fd77da2c4b9f3c6d25f563fca66 /modules/extras.py
parent29a2933e23e68900bbae741a98439d0c9d6f26f6 (diff)
parent303b75c1490e8f9e146ba207f3425a7f0317cb12 (diff)
downloadstable-diffusion-webui-gfx803-e05e46aa3f3b2230affe399372ad96d8736e4010.tar.gz
stable-diffusion-webui-gfx803-e05e46aa3f3b2230affe399372ad96d8736e4010.tar.bz2
stable-diffusion-webui-gfx803-e05e46aa3f3b2230affe399372ad96d8736e4010.zip
Merge branch 'master' into master
Diffstat (limited to 'modules/extras.py')
-rw-r--r--modules/extras.py4
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 = ""