diff --git a/direct/nn/transformers/mri_transformers.py b/direct/nn/transformers/mri_transformers.py
index dd2ab81f..65c2f06e 100644
--- a/direct/nn/transformers/mri_transformers.py
+++ b/direct/nn/transformers/mri_transformers.py
@@ -78,6 +78,7 @@ def __init__(
         locality_strength: float = 1.0,
         use_pos_embedding: bool = True,
         normalized: bool = True,
+        **kwargs,
     ) -> None:
         """Inits :class:`ImageDomainMRIViT2D`.
 
@@ -231,6 +232,7 @@ def __init__(
         locality_strength: float = 1.0,
         use_pos_embedding: bool = True,
         normalized: bool = True,
+        **kwargs,
     ) -> None:
         """Inits :class:`ImageDomainMRIViT3D`.
 
@@ -386,6 +388,7 @@ def __init__(
         use_pos_embedding: bool = True,
         normalized: bool = True,
         compute_per_coil: bool = True,
+        **kwargs,
     ) -> None:
         """Inits :class:`KSpaceDomainMRIViT2D`.
 
@@ -575,6 +578,7 @@ def __init__(
         use_pos_embedding: bool = True,
         normalized: bool = True,
         compute_per_coil: bool = True,
+        **kwargs,
     ) -> None:
         """Inits :class:`KSpaceDomainMRIViT3D`.