Update generator.py

generator.py

@@ -153,8 +153,9 @@ class RetroArtConverter:
                 if image.mode != 'RGB':
                     image = image.convert('RGB')
                 
-                # ZoeDetector handles everything internally - just pass the image
-                depth_image = self.zoe_depth(image)
+                # CRITICAL: ZoeDetector must be called with torch.no_grad()
+                with torch.no_grad():
+                    depth_image = self.zoe_depth(image)
                 
                 # Use .width and .height properties (always Python ints, not numpy types)
                 print(f"[DEPTH] Zoe depth map generated: {image.width}x{image.height}")