1580158
/
kohya_ss
mirror of https://github.com/bmaltais/kohya_ss
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'sdxl' of https://github.com/kohya-ss/sd-scripts into dev2

pull/1223/head
bmaltais 2023-07-12 08:41:00 -04:00
parent b602aa2ec0 3c67e595b8
commit dfe847071f
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
sdxl_train.py
View File

@ -362,8 +362,7 @@ def train(args):
loss_total = 0
for step, batch in enumerate(train_dataloader):
current_step.value = global_step
# with accelerator.accumulate(training_models[0]): # 複数モデルに対応していない模様だがとりあえずこうしておく
if True:
with accelerator.accumulate(training_models[0]): # 複数モデルに対応していない模様だがとりあえずこうしておく
if "latents" in batch and batch["latents"] is not None:
latents = batch["latents"].to(accelerator.device).to(dtype=weight_dtype)
else: