From e304e4db354906493f9e22866b8fcee5f403a829 Mon Sep 17 00:00:00 2001 From: MickeyCHAN <76671016+danyow-cheung@users.noreply.github.com> Date: Tue, 5 Mar 2024 21:41:23 +0800 Subject: [PATCH] [hotfix] fix sd vit import error (#5420) * fix import error * Update dpt_depth.py --------- Co-authored-by: binmakeswell --- examples/images/diffusion/ldm/modules/midas/midas/dpt_depth.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/images/diffusion/ldm/modules/midas/midas/dpt_depth.py b/examples/images/diffusion/ldm/modules/midas/midas/dpt_depth.py index 74871e8b1..fbb0e0563 100644 --- a/examples/images/diffusion/ldm/modules/midas/midas/dpt_depth.py +++ b/examples/images/diffusion/ldm/modules/midas/midas/dpt_depth.py @@ -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):