mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2024-06-07 21:20:49 +00:00
fix ci
This commit is contained in:
parent
fffc902698
commit
8828c9ecc5
@ -1,6 +1,5 @@
|
||||
from contextlib import contextmanager, nullcontext
|
||||
import torch
|
||||
from torch.utils.weak import WeakIdKeyDictionary
|
||||
from modules import devices, shared, patches
|
||||
|
||||
module_in_gpu = None
|
||||
@ -86,7 +85,7 @@ class RTTensorMoverPatches:
|
||||
self.stash[id(after_calc_event)] = (weight, bias, after_calc_event)
|
||||
|
||||
to_remove = []
|
||||
for k, (w, b, e) in self.stash.items():
|
||||
for k, (_, _, e) in self.stash.items():
|
||||
if e.query():
|
||||
to_remove.append(k)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user