diff --git a/__init__.py b/__init__.py
index 4191b3c..d59966f 100644
--- a/__init__.py
+++ b/__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)