github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-24 14:33:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
* fix import error

* Update dpt_depth.py

---------

Co-authored-by: binmakeswell <binmakeswell@gmail.com>
This commit is contained in:
MickeyCHAN 2024-03-05 21:41:23 +08:00 committed by GitHub
parent 070df689e6
commit e304e4db35
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
examples/images/diffusion/ldm/modules/midas/midas/dpt_depth.py
View File

@ -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):