Mirror/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2024-06-07 21:20:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix ci

Browse Source
This commit is contained in:
wfjsw 2024-02-20 21:08:07 -06:00
parent fffc902698
commit 8828c9ecc5
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules/lowvram.py
View File

@ -1,6 +1,5 @@
from contextlib import contextmanager, nullcontext from contextlib import contextmanager, nullcontext
import torch import torch
from torch.utils.weak import WeakIdKeyDictionary
from modules import devices, shared, patches from modules import devices, shared, patches
module_in_gpu = None module_in_gpu = None
@ -86,7 +85,7 @@ class RTTensorMoverPatches:
self.stash[id(after_calc_event)] = (weight, bias, after_calc_event) self.stash[id(after_calc_event)] = (weight, bias, after_calc_event)
to_remove = [] to_remove = []
for k, (w, b, e) in self.stash.items(): for k, (_, _, e) in self.stash.items():
if e.query(): if e.query():
to_remove.append(k) to_remove.append(k)