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:
parent
07ea52542e
commit
e0f97b6fef
|
|
@ -1 +1 @@
|
||||||
version = '2.0.2'
|
version = '2.0.3'
|
||||||
|
|
|
||||||
|
|
@ -1,4 +1,5 @@
|
||||||
import threading
|
import threading
|
||||||
|
import torch
|
||||||
|
|
||||||
|
|
||||||
buffer = []
|
buffer = []
|
||||||
|
|
@ -33,6 +34,7 @@ def worker():
|
||||||
def progressbar(number, text):
|
def progressbar(number, text):
|
||||||
outputs.append(['preview', (number, text, None)])
|
outputs.append(['preview', (number, text, None)])
|
||||||
|
|
||||||
|
@torch.no_grad()
|
||||||
def handler(task):
|
def handler(task):
|
||||||
prompt, negative_prompt, style_selections, performance_selction, \
|
prompt, negative_prompt, style_selections, performance_selction, \
|
||||||
aspect_ratios_selction, image_number, image_seed, sharpness, \
|
aspect_ratios_selction, image_number, image_seed, sharpness, \
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue