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

pop weights if available #1247

Browse Source
This commit is contained in:
Kohya S
2024-04-21 17:45:29 +09:00
parent 52652cba1a
commit 0540c33aca
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
train_network.py
View File

@@ -481,7 +481,8 @@ class NetworkTrainer:
if not isinstance(model, type(accelerator.unwrap_model(network))): if not isinstance(model, type(accelerator.unwrap_model(network))):
remove_indices.append(i) remove_indices.append(i)
for i in reversed(remove_indices): for i in reversed(remove_indices):
weights.pop(i) if len(weights) > i:
weights.pop(i)
# print(f"save model hook: {len(weights)} weights will be saved") # print(f"save model hook: {len(weights)} weights will be saved")
def load_model_hook(models, input_dir): def load_model_hook(models, input_dir):