diff --git a/mmagic/models/archs/lora.py b/mmagic/models/archs/lora.py
index 066a13cbd4..a07a7e230e 100644
--- a/mmagic/models/archs/lora.py
+++ b/mmagic/models/archs/lora.py
@@ -219,7 +219,7 @@ def forward_lora_mapping(self, x: Tensor) -> Tensor:
                 mapping_out = self.scale * self.lora_mapping(x)
         return mapping_out
 
-    def forward(self, x: Tensor) -> Tensor:
+    def forward(self, x: Tensor, *args, **kwargs) -> Tensor:
         """Forward and add LoRA mapping.
 
         Args: