added outpainting on injected frames and shapes preprocessor

2025-06-20 23:57:07 +02:00
parent 8b146a8d7b
commit d0a32c67a0
5 changed files with 276 additions and 80 deletions
--- a/preprocessing/depth_anything_v2/depth.py
+++ b/preprocessing/depth_anything_v2/depth.py
@@ -26,7 +26,8 @@ class DepthV2Annotator:
        self.model.load_state_dict(
            torch.load(
                pretrained_model,
-                map_location=self.device
+                map_location=self.device,
+                weights_only=True
            )
        )
        self.model.eval()