Commit 3d58510f authored by Muhammad Rizqi Nur's avatar Muhammad Rizqi Nur
Browse files

Fix dataset still being loaded even when training will be skipped

parent a07f054c

modules/hypernetworks/hypernetwork.py

+1 −1
Original line number Diff line number Diff line
@@ -364,7 +364,7 @@ def train_hypernetwork(hypernetwork_name, learn_rate, batch_size, data_root, log 
    checkpoint = sd_models.select_checkpoint()



    ititial_step = hypernetwork.step or 0

    if ititial_step > steps:

    if ititial_step >= steps:

        shared.state.textinfo = f"Model has already been trained beyond specified max steps"

        return hypernetwork, filename

modules/textual_inversion/textual_inversion.py

+1 −1
Original line number Diff line number Diff line
@@ -262,7 +262,7 @@ def train_embedding(embedding_name, learn_rate, batch_size, data_root, log_direc 
    checkpoint = sd_models.select_checkpoint()



    ititial_step = embedding.step or 0

    if ititial_step > steps:

    if ititial_step >= steps:

        shared.state.textinfo = f"Model has already been trained beyond specified max steps"

        return embedding, filename

CRA Git | Maintained and supported by SUSTech CRA and CCSE