Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: use v1 models for large and base versions

Browse Source
This commit is contained in:
blessedcoolant 2024-07-25 17:24:12 +05:30
parent 4166c756ce
commit e5d9ca013e
4 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/backend/image_util/depth_anything/__init__.py
View File

@ -14,8 +14,8 @@ config = get_config()
logger = InvokeAILogger.get_logger(config=config) logger = InvokeAILogger.get_logger(config=config)
DEPTH_ANYTHING_MODELS = { DEPTH_ANYTHING_MODELS = {
"large": "https://huggingface.co/depth-anything/Depth-Anything-V2-Large/resolve/main/depth_anything_v2_vitl.pth?download=true", "large": "https://huggingface.co/spaces/LiheYoung/Depth-Anything/resolve/main/checkpoints/depth_anything_vitl14.pth?download=true",
"base": "https://huggingface.co/depth-anything/Depth-Anything-V2-Base/resolve/main/depth_anything_v2_vitb.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", "small": "https://huggingface.co/depth-anything/Depth-Anything-V2-Small/resolve/main/depth_anything_v2_vits.pth?download=true",
} }

0
invokeai/backend/image_util/depth_anything/v2/utils/blocks.py → invokeai/backend/image_util/depth_anything/utils/blocks.py
View File

0
invokeai/backend/image_util/depth_anything/v2/utils/transform.py → invokeai/backend/image_util/depth_anything/utils/transform.py
View File

4
invokeai/backend/image_util/depth_anything/v2/dpt.py
View File

@ -9,9 +9,9 @@ import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
from torchvision.transforms import Compose 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.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): def _make_fusion_block(features, use_bn, size=None):