just no grad everything to prevent mistakes (#359)

* just no grad everything to prevent mistakes

* just no grad everything to prevent mistakes
This commit is contained in:
lllyasviel 2023-09-13 03:07:47 -07:00 committed by GitHub
parent 07ea52542e
commit e0f97b6fef
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 1 deletions

View File

@ -1 +1 @@
version = '2.0.2'
version = '2.0.3'

View File

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