feat(transformers): Add depth estimation pipe

wcrzlh · Jan 22, 2025 · 0b85425 · 0b85425
1 parent a70a4b9
commit 0b85425
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mindone/transformers/models/depth_anything/modeling_depth_anything.py b/mindone/transformers/models/depth_anything/modeling_depth_anything.py
@@ -285,7 +285,7 @@ def __init__(self, config):
 
         self.convs = nn.CellList()
         for channel in config.neck_hidden_sizes:
-            self.convs.append(nn.Conv2d(channel, config.fusion_hidden_size, kernel_size=3, padding=1, has_bias=False))
+            self.convs.append(nn.Conv2d(channel, config.fusion_hidden_size, kernel_size=3, padding=1, has_bias=False, pad_mode="same"))
 
         # fusion
         self.fusion_stage = DepthAnythingFeatureFusionStage(config)