Browse Source
add2
Signed-off-by: xujinling <jinling.xu@zilliz.com>
main
xujinling
2 years ago
1 changed files with
2 additions and
2 deletions
-
__init__.py
|
|
@ -15,6 +15,6 @@ |
|
|
|
from .video_swin_transformer import VideoSwinTransformer |
|
|
|
|
|
|
|
|
|
|
|
def video_swin_transformer(model_name: str, modality: str, **kwargs): |
|
|
|
return VideoSwinTransformer(model_name, modality, **kwargs) |
|
|
|
def video_swin_transformer(**kwargs): |
|
|
|
return VideoSwinTransformer(**kwargs) |
|
|
|
|
|
|
|