Fixed bug that prevented using schnell training adapter

2025-03-19 10:25:12 -06:00
parent b829983b16
commit bbfd6ef0fe
2 changed files with 7 additions and 3 deletions
--- a/toolkit/lora_special.py
+++ b/toolkit/lora_special.py
@@ -209,7 +209,9 @@ class LoRASpecialNetwork(ToolkitNetworkMixin, LoRANetwork):
            ignore_if_contains = []
        self.ignore_if_contains = ignore_if_contains
        self.transformer_only = transformer_only
-        self.base_model_ref = weakref.ref(base_model)
+        self.base_model_ref = None
+        if base_model is not None:
+            self.base_model_ref = weakref.ref(base_model)

        self.only_if_contains: Union[List, None] = only_if_contains