diff --git a/invokeai/backend/image_util/depth_anything/__init__.py b/invokeai/backend/image_util/depth_anything/__init__.py index 8052765d93..5ed9b2e20b 100644 --- a/invokeai/backend/image_util/depth_anything/__init__.py +++ b/invokeai/backend/image_util/depth_anything/__init__.py @@ -14,8 +14,8 @@ config = get_config() logger = InvokeAILogger.get_logger(config=config) DEPTH_ANYTHING_MODELS = { - "large": "https://huggingface.co/depth-anything/Depth-Anything-V2-Large/resolve/main/depth_anything_v2_vitl.pth?download=true", - "base": "https://huggingface.co/depth-anything/Depth-Anything-V2-Base/resolve/main/depth_anything_v2_vitb.pth?download=true", + "large": "https://huggingface.co/spaces/LiheYoung/Depth-Anything/resolve/main/checkpoints/depth_anything_vitl14.pth?download=true", + "base": "https://huggingface.co/spaces/LiheYoung/Depth-Anything/resolve/main/checkpoints/depth_anything_vitb14.pth?download=true", "small": "https://huggingface.co/depth-anything/Depth-Anything-V2-Small/resolve/main/depth_anything_v2_vits.pth?download=true", } diff --git a/invokeai/backend/image_util/depth_anything/v2/utils/blocks.py b/invokeai/backend/image_util/depth_anything/utils/blocks.py similarity index 100% rename from invokeai/backend/image_util/depth_anything/v2/utils/blocks.py rename to invokeai/backend/image_util/depth_anything/utils/blocks.py diff --git a/invokeai/backend/image_util/depth_anything/v2/utils/transform.py b/invokeai/backend/image_util/depth_anything/utils/transform.py similarity index 100% rename from invokeai/backend/image_util/depth_anything/v2/utils/transform.py rename to invokeai/backend/image_util/depth_anything/utils/transform.py diff --git a/invokeai/backend/image_util/depth_anything/v2/dpt.py b/invokeai/backend/image_util/depth_anything/v2/dpt.py index 3427e963fa..d9146ee672 100644 --- a/invokeai/backend/image_util/depth_anything/v2/dpt.py +++ b/invokeai/backend/image_util/depth_anything/v2/dpt.py @@ -9,9 +9,9 @@ import torch.nn as nn import torch.nn.functional as F from torchvision.transforms import Compose +from invokeai.backend.image_util.depth_anything.utils.blocks import FeatureFusionBlock, _make_scratch +from invokeai.backend.image_util.depth_anything.utils.transform import NormalizeImage, PrepareForNet, Resize from invokeai.backend.image_util.depth_anything.v2.dinov2 import DINOv2 -from invokeai.backend.image_util.depth_anything.v2.utils.blocks import FeatureFusionBlock, _make_scratch -from invokeai.backend.image_util.depth_anything.v2.utils.transform import NormalizeImage, PrepareForNet, Resize def _make_fusion_block(features, use_bn, size=None):