VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-08 22:35:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add VAE to meatada, add no_metadata option

Browse Source
This commit is contained in:
Kohya S
2023-01-11 23:12:18 +09:00
parent 895a599d34
commit e4f9b2b715
3 changed files with 59 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
networks/lora.py
View File

@@ -178,6 +178,9 @@ class LoRANetwork(torch.nn.Module):
return self.parameters()
def save_weights(self, file, dtype, metadata):
if len(metadata) == 0:
metadata = None
state_dict = self.state_dict()
if dtype is not None:
@@ -191,5 +194,6 @@ class LoRANetwork(torch.nn.Module):
save_file(state_dict, file, metadata)
else:
torch.save(state_dict, file)
with zipfile.ZipFile(file, "w") as zipf:
zipf.writestr(LoRANetwork.METADATA_FILENAME, json.dumps(metadata))
if metadata is not None:
with zipfile.ZipFile(file, "w") as zipf:
zipf.writestr(LoRANetwork.METADATA_FILENAME, json.dumps(metadata))