get lowvram from sd_model (#404)

2024-02-18 22:47:23 -06:00 · 2024-02-18 22:47:23 -06:00 · 4c3ec2e5fb
parent fe02e05d85
commit 4c3ec2e5fb
1 changed files with 1 additions and 1 deletions
--- a/scripts/animatediff_mm.py
+++ b/scripts/animatediff_mm.py
@ -148,7 +148,7 @@ class AnimateDiffMM:

        AnimateDiffMM.mm_injected = False
        logger.info(f"Removal finished.")
-        if shared.cmd_opts.lowvram:
+        if sd_model.lowvram:
            self.unload()