diff --git a/networks/lora_flux.py b/networks/lora_flux.py index efc7847e..07a80f0b 100644 --- a/networks/lora_flux.py +++ b/networks/lora_flux.py @@ -604,7 +604,7 @@ class LoRANetwork(torch.nn.Module): return info def load_state_dict(self, state_dict, strict=True): - # override to convert original weight to splitted qkv weight + # override to convert original weight to split qkv if not self.split_qkv: return super().load_state_dict(state_dict, strict)