From e0f97b6fefcacf6bce40c07ea81bd40f0b994ddd Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Wed, 13 Sep 2023 03:07:47 -0700 Subject: [PATCH] just no grad everything to prevent mistakes (#359) * just no grad everything to prevent mistakes * just no grad everything to prevent mistakes --- fooocus_version.py | 2 +- modules/async_worker.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/fooocus_version.py b/fooocus_version.py index 8e95544e..e83c613b 100644 --- a/fooocus_version.py +++ b/fooocus_version.py @@ -1 +1 @@ -version = '2.0.2' +version = '2.0.3' diff --git a/modules/async_worker.py b/modules/async_worker.py index 57818a3e..76b54761 100644 --- a/modules/async_worker.py +++ b/modules/async_worker.py @@ -1,4 +1,5 @@ import threading +import torch buffer = [] @@ -33,6 +34,7 @@ def worker(): def progressbar(number, text): outputs.append(['preview', (number, text, None)]) + @torch.no_grad() def handler(task): prompt, negative_prompt, style_selections, performance_selction, \ aspect_ratios_selction, image_number, image_seed, sharpness, \