"lib/runtime/examples/service_metrics/src/lib.rs" did not exist on "cec8248d53d4da0b07d878e5167a49f6d0d88fa7"
Unverified Commit e304e4db authored by MickeyCHAN's avatar MickeyCHAN Committed by GitHub
Browse files

[hotfix] fix sd vit import error (#5420)



* fix import error

* Update dpt_depth.py

---------
Co-authored-by: default avatarbinmakeswell <binmakeswell@gmail.com>
parent 070df689
...@@ -2,7 +2,8 @@ import torch ...@@ -2,7 +2,8 @@ import torch
import torch.nn as nn import torch.nn as nn
from .base_model import BaseModel from .base_model import BaseModel
from .blocks import FeatureFusionBlock_custom, Interpolate, _make_encoder, forward_vit from .blocks import FeatureFusionBlock_custom, Interpolate, _make_encoder
from .vit import forward_vit
def _make_fusion_block(features, use_bn): def _make_fusion_block(features, use_bn):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment