This commit is contained in:
aria1th 2023-12-08 01:51:13 +09:00
parent f166868df6
commit 74fb119452

View file

@ -71,7 +71,7 @@ class DeepCacheSession:
Registers cache
"""
CACHE_LAST["timestep"].add(timestep)
assert h is not None, f"Cannot cache None"
assert h is not None, "Cannot cache None"
# maybe move to cpu and load later for low vram?
CACHE_LAST["last"] = h
CACHE_LAST[f"timestep_{timestep}"] = h