Skip to content

Commit e5d9ca0

Browse files
fix: use v1 models for large and base versions
1 parent 4166c75 commit e5d9ca0

File tree

4 files changed

+4
-4
lines changed

4 files changed

+4
-4
lines changed

invokeai/backend/image_util/depth_anything/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,8 @@
1414
logger = InvokeAILogger.get_logger(config=config)
1515

1616
DEPTH_ANYTHING_MODELS = {
17-
"large": "https://huggingface.co/depth-anything/Depth-Anything-V2-Large/resolve/main/depth_anything_v2_vitl.pth?download=true",
18-
"base": "https://huggingface.co/depth-anything/Depth-Anything-V2-Base/resolve/main/depth_anything_v2_vitb.pth?download=true",
17+
"large": "https://huggingface.co/spaces/LiheYoung/Depth-Anything/resolve/main/checkpoints/depth_anything_vitl14.pth?download=true",
18+
"base": "https://huggingface.co/spaces/LiheYoung/Depth-Anything/resolve/main/checkpoints/depth_anything_vitb14.pth?download=true",
1919
"small": "https://huggingface.co/depth-anything/Depth-Anything-V2-Small/resolve/main/depth_anything_v2_vits.pth?download=true",
2020
}
2121

invokeai/backend/image_util/depth_anything/v2/dpt.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,9 +9,9 @@
99
import torch.nn.functional as F
1010
from torchvision.transforms import Compose
1111

12+
from invokeai.backend.image_util.depth_anything.utils.blocks import FeatureFusionBlock, _make_scratch
13+
from invokeai.backend.image_util.depth_anything.utils.transform import NormalizeImage, PrepareForNet, Resize
1214
from invokeai.backend.image_util.depth_anything.v2.dinov2 import DINOv2
13-
from invokeai.backend.image_util.depth_anything.v2.utils.blocks import FeatureFusionBlock, _make_scratch
14-
from invokeai.backend.image_util.depth_anything.v2.utils.transform import NormalizeImage, PrepareForNet, Resize
1515

1616

1717
def _make_fusion_block(features, use_bn, size=None):

0 commit comments

Comments
 (0)