Add torch_gc to state.nextjob, vae and upscale

2026-09-20 09:38:23 +02:00 · 2024-06-20 14:47:30 +03:00
parent eb6e3c337c
commit 092a326c09
4 changed files with 11 additions and 3 deletions
@@ -41,10 +41,12 @@ class State:
        log.debug(f'Requested {"pause" if self.paused else "continue"}')

    def nextjob(self):
+        import modules.devices
        self.do_set_current_image()
        self.job_no += 1
        self.sampling_step = 0
        self.current_image_sampling_step = 0
+        modules.devices.torch_gc()

    def dict(self):
        obj = {