Spaces:

developy
/

ApDepth

Running

developy commited on 14 days ago

Commit

5827a9c

verified ·

1 Parent(s): 99f453c

Update apdepth/marigold_pipeline.py

Files changed (1) hide show

apdepth/marigold_pipeline.py CHANGED Viewed

@@ -146,9 +146,9 @@ class MarigoldPipeline(DiffusionPipeline):
         self._fft_masks = {}
         da2_config = {
-            'encoder': 'vits',  # 'vits', 'vitb', 'vitl', 'vitg'
-            'features': 64,
-            'out_channels': [48, 96, 192, 384],
         }
         # 初始化 DA2 模型

         self._fft_masks = {}
         da2_config = {
+            'encoder': 'vitb',  # 'vits', 'vitb', 'vitl', 'vitg'
+            'features': 128,
+            'out_channels': [96, 192, 384, 768],
         }
         # 初始化 DA2 模型