Fix full modules(manual cherry pick)

KohakuBlueleaf · Dec 14, 2023 · f86f757 · f86f757
1 parent aa66fcb
commit f86f757
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lycoris/modules/full.py b/lycoris/modules/full.py
@@ -53,6 +53,7 @@ def __init__(
 
     def apply_to(self, **kwargs):
         self.org_forward = self.org_module[0].forward
+        self.org_module[0].forward = self.forward
         self.weight.data.copy_(self.org_module[0].weight.data)
         self.org_weight = [self.org_module[0].weight.data.cpu().clone()]
         delattr(self.org_module[0], "weight")