mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-09 06:45:09 +00:00
typo修正、stepをglobal_stepに修正、バグ修正
This commit is contained in:
@@ -340,8 +340,8 @@ def train(args):
|
||||
|
||||
loss_total = 0
|
||||
for step, batch in enumerate(train_dataloader):
|
||||
train_dataset_group.set_current_step(global_step)
|
||||
with accelerator.accumulate(text_encoder):
|
||||
train_dataset_group.set_current_step(step + 1)
|
||||
with torch.no_grad():
|
||||
if "latents" in batch and batch["latents"] is not None:
|
||||
latents = batch["latents"].to(accelerator.device)
|
||||
|
||||
Reference in New Issue
Block a user