update support for latest diffusers with lora (open-mmlab#2067)

liuwenran · Nov 10, 2023 · ef68f11 · ef68f11
1 parent eb3b3e4
commit ef68f11
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: