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
import torch.nn as nn
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):
......
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