From 8fb9c57ed62dcef721d50c1eeb9c20f65c509215 Mon Sep 17 00:00:00 2001 From: AUTOMATIC <16777216c@gmail.com> Date: Sun, 11 Sep 2022 23:24:24 +0300 Subject: add half() supporrt for CLIP interrogation --- modules/extras.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'modules/extras.py') 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 = "" -- cgit v1.2.3