This commit is contained in:
Kohya S
2024-08-24 16:37:49 +09:00
parent cf689e7aa6
commit 5639c2adc0

View File

@@ -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)