From cdcb21e13f3ef7d333cd38bea132b3e24cc11cf3 Mon Sep 17 00:00:00 2001 From: zengyh1900 Date: Fri, 10 Nov 2023 21:07:56 +0800 Subject: [PATCH] update support for latest diffusers with lora --- 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: