Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-08 05:48:42 +02:00
Code Issues Packages Projects Releases Wiki Activity

update paths

Browse Source
This commit is contained in:
Vladimir Mandic
2023-01-08 17:04:51 -05:00
parent 90266b13fb
commit 91d0fc6c95
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
config.json
+2 -2
View File
@@ -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,
modules/textual_inversion/textual_inversion.py
+2 -2
View File
@@ -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
})