hysts HF staff commited on
Commit
5520f5c
1 Parent(s): fd5cb5f
Files changed (1) hide show
  1. trainer.py +1 -1
trainer.py CHANGED
@@ -100,7 +100,7 @@ class Trainer:
100
  --max_train_steps={n_steps}
101
  '''
102
  if fp16:
103
- command += ' --mixed_precision fp16 '
104
  if use_8bit_adam:
105
  command += ' --use_8bit_adam'
106
  if train_text_encoder:
 
100
  --max_train_steps={n_steps}
101
  '''
102
  if fp16:
103
+ command += ' --mixed_precision fp16'
104
  if use_8bit_adam:
105
  command += ' --use_8bit_adam'
106
  if train_text_encoder: