This commit is contained in:
wfjsw 2024-02-11 22:06:59 -06:00
parent 8278ad01e2
commit 5d69b1e803

View File

@ -558,7 +558,7 @@ def setup_for_low_vram(sd_model, use_medvram):
# install hooks for bits of third model # install hooks for bits of third model
if stream_impl is not None and stream_wrapper is not None: if stream_impl is not None and stream_wrapper is not None:
mp = DiffSmartModelMover.register(diff_model, max_prefetch=70) mp = DiffSmartModelMover.register(diff_model, vram_allowance=512, max_prefetch=70)
mp.install() mp.install()
mp.preload() mp.preload()
else: else: