Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

use cudnn workaround

Browse Source
This commit is contained in:
Vladimir Mandic
2023-05-11 22:24:12 -04:00
parent df2cfd5100
commit 1943bfea88
5 changed files with 18 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing.py
+1 -3
View File
@@ -538,7 +538,7 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
assert len(p.prompt) > 0
else:
assert p.prompt is not None
devices.torch_gc()
# devices.torch_gc() # TODO: gc
seed = get_fixed_seed(p.seed)
subseed = get_fixed_seed(p.subseed)
modules.sd_hijack.model_hijack.apply_circular(p.tiling)
@@ -672,9 +672,7 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
info=infotext(n, i)
p.restore_faces = orig
images.save_image(Image.fromarray(x_sample), p.outpath_samples, "", seeds[i], prompts[i], opts.samples_format, info=info, p=p, suffix="-before-face-restoration")
devices.torch_gc()
x_sample = modules.face_restoration.restore_faces(x_sample)
devices.torch_gc()
image = Image.fromarray(x_sample)
if p.scripts is not None:
pp = scripts.PostprocessImageArgs(image)