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

resume from huggingface repository

Browse Source
This commit is contained in:
ddPn08
2023-03-30 23:36:42 +09:00
parent a7d302e196
commit b5ff4e816f
8 changed files with 139 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
train_textual_inversion.py
View File

@@ -304,9 +304,7 @@ def train(args):
text_encoder.to(weight_dtype)
# resumeする
if args.resume is not None:
print(f"resume training from state: {args.resume}")
accelerator.load_state(args.resume)
train_util.resume(accelerator, args)
# epoch数を計算する
num_update_steps_per_epoch = math.ceil(len(train_dataloader) / args.gradient_accumulation_steps)