Vace Outpainting, new Depth extractor

2025-06-19 15:59:47 +02:00
parent 188225cdd7
commit 4d202db319
27 changed files with 2023 additions and 180 deletions
--- a/preprocessing/midas/vit.py
+++ b/preprocessing/midas/vit.py
@@ -498,6 +498,7 @@ def _make_pretrained_vitb_rn50_384(pretrained,
                                   hooks=None,
                                   use_vit_only=False):
    model = timm.create_model('vit_base_resnet50_384', pretrained=pretrained)
+    # model = timm.create_model('vit_base_r50_s16_384.orig_in21k_ft_in1k', pretrained=pretrained)

    hooks = [0, 1, 8, 11] if hooks is None else hooks
    return _make_vit_b_rn50_backbone(