Spaces:

hysts
/

LoRA-SD-training

Build error

hysts HF staff commited on Dec 14, 2022

Commit

5520f5c

•

1 Parent(s): fd5cb5f

Clean up

Files changed (1) hide show

trainer.py CHANGED Viewed

@@ -100,7 +100,7 @@ class Trainer:
           --max_train_steps={n_steps}
         '''
         if fp16:
-            command += ' --mixed_precision fp16 '
         if use_8bit_adam:
             command += ' --use_8bit_adam'
         if train_text_encoder:

           --max_train_steps={n_steps}
         '''
         if fp16:
+            command += ' --mixed_precision fp16'
         if use_8bit_adam:
             command += ' --use_8bit_adam'
         if train_text_encoder: