Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Vladimir Mandic
2023-01-09 17:38:58 -05:00
parent 02dddd2feb
commit 5754401e50
2 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+1 -1
View File
@@ -1,3 +1,3 @@
# Stable Diffusion - Automatic
Custom fork of <https://github.com/AUTOMATIC1111/stable-diffusion-webui>
Heavily opinionated custom fork of <https://github.com/AUTOMATIC1111/stable-diffusion-webui>
modules/textual_inversion/textual_inversion.py
+2 -4
View File
@@ -255,13 +255,11 @@ def write_loss(log_directory, filename, step, epoch_len, values):
if step % shared.opts.training_write_csv_every != 0:
return
write_csv_header = False if os.path.exists(os.path.join(log_directory, filename)) else True
with open(os.path.join(log_directory, filename), "a+", newline='') as fout:
with open(os.path.join(log_directory, filename), "w", newline='') as fout:
csv_writer = csv.DictWriter(fout, fieldnames=["step", "epoch", "epoch_step", *(values.keys())])
if write_csv_header:
csv_writer.writeheader()
csv_writer.writeheader()
epoch = (step - 1) // epoch_len
epoch_step = (step - 1) % epoch_len