From ef68f1176af87470e7aa0c9424d298c12740b570 Mon Sep 17 00:00:00 2001 From: Yanhong Zeng Date: Fri, 10 Nov 2023 21:14:08 +0800 Subject: [PATCH] update support for latest diffusers with lora (#2067) --- mmagic/models/archs/lora.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: