diff --git a/config.json b/config.json index 7eb91f217..19443ad3e 100644 --- a/config.json +++ b/config.json @@ -54,9 +54,9 @@ "dataset_filename_word_regex": "", "dataset_filename_join_string": " ", "training_image_repeats_per_epoch": 1, - "training_write_csv_every": 100.0, + "training_write_csv_every": 10.0, "training_xattention_optimizations": false, - "sd_model_checkpoint": "sd-v15-runwayml-inpainting.ckpt [3e16efc8]", + "sd_model_checkpoint": "sd-v15-runwayml.ckpt [81761151]", "sd_checkpoint_cache": 0, "sd_vae": "vae-ft-mse-840000-ema-pruned", "sd_vae_as_default": false, diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py index 45882ed68..590d92d95 100644 --- a/modules/textual_inversion/textual_inversion.py +++ b/modules/textual_inversion/textual_inversion.py @@ -266,7 +266,7 @@ def train_embedding(embedding_name, learn_rate, batch_size, gradient_step, data_ filename = os.path.join(shared.cmd_opts.embeddings_dir, f'{embedding_name}.pt') - log_directory = os.path.join(log_directory, datetime.datetime.now().strftime("%Y-%m-%d"), embedding_name) + log_directory = os.path.join(log_directory, embedding_name) unload = shared.opts.unload_models_when_training if save_embedding_every > 0: @@ -421,7 +421,7 @@ def train_embedding(embedding_name, learn_rate, batch_size, gradient_step, data_ save_embedding(embedding, optimizer, checkpoint, embedding_name_every, last_saved_file, remove_cached_checksum=True) embedding_yet_to_be_embedded = True - write_loss(log_directory, "textual_inversion_loss.csv", embedding.step, steps_per_epoch, { + write_loss(log_directory, "train.csv", embedding.step, steps_per_epoch, { "loss": f"{loss_step:.7f}", "learn_rate": scheduler.learn_rate })