summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorVolpeon <git@volpeon.ink>2023-01-11 21:57:43 +0100
committerVolpeon <git@volpeon.ink>2023-01-11 21:57:43 +0100
commitd12634cb211e10223ad77a43a562d80678b2d662 (patch)
tree47b0678a33617e49090ae6690c8b83bc867d3b34
parentTI: Use grad clipping from LoRA #104 (diff)
downloadtextual-inversion-diff-d12634cb211e10223ad77a43a562d80678b2d662.tar.gz
textual-inversion-diff-d12634cb211e10223ad77a43a562d80678b2d662.tar.bz2
textual-inversion-diff-d12634cb211e10223ad77a43a562d80678b2d662.zip
Heck
-rw-r--r--train_ti.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/train_ti.py b/train_ti.py
index 1054a5d..e6c437e 100644
--- a/train_ti.py
+++ b/train_ti.py
@@ -974,7 +974,7 @@ def main():
974 974
975 try: 975 try:
976 for epoch in range(num_epochs): 976 for epoch in range(num_epochs):
977 if accelerator.is_main_process and False: 977 if accelerator.is_main_process:
978 if epoch % args.sample_frequency == 0: 978 if epoch % args.sample_frequency == 0:
979 checkpointer.save_samples(global_step + global_step_offset, args.sample_steps) 979 checkpointer.save_samples(global_step + global_step_offset, args.sample_steps)
980 980