Skip to content

Commit 3d58510

Browse files
committed
Fix dataset still being loaded even when training will be skipped
1 parent a07f054 commit 3d58510

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

modules/hypernetworks/hypernetwork.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -364,7 +364,7 @@ def train_hypernetwork(hypernetwork_name, learn_rate, batch_size, data_root, log
364364
checkpoint = sd_models.select_checkpoint()
365365

366366
ititial_step = hypernetwork.step or 0
367-
if ititial_step > steps:
367+
if ititial_step >= steps:
368368
shared.state.textinfo = f"Model has already been trained beyond specified max steps"
369369
return hypernetwork, filename
370370

modules/textual_inversion/textual_inversion.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -262,7 +262,7 @@ def train_embedding(embedding_name, learn_rate, batch_size, data_root, log_direc
262262
checkpoint = sd_models.select_checkpoint()
263263

264264
ititial_step = embedding.step or 0
265-
if ititial_step > steps:
265+
if ititial_step >= steps:
266266
shared.state.textinfo = f"Model has already been trained beyond specified max steps"
267267
return embedding, filename
268268

0 commit comments

Comments
 (0)