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, \