Bugfix: fix offloading

semjon00 · semjon00 · commit bc3755ec78cd · 2023-07-19T10:53:41.000+03:00
diff --git a/src/depthmap_generation.py b/src/depthmap_generation.py
@@ -248,7 +248,7 @@ def reload(self):
         """Undoes offload"""
         if self.offloaded:
             self.move_models_to(self.device)
-            self.offloaded = True
+            self.offloaded = False
 
     def move_models_to(self, device):
         if self.depth_model is not None:
