nupurkmr9 commited on
Commit
7403db4
1 Parent(s): 751f591

Update trainer.py

Browse files
Files changed (1) hide show
  1. trainer.py +1 -1
trainer.py CHANGED
@@ -112,7 +112,7 @@ class Trainer:
112
  if use_8bit_adam:
113
  command += ' --use_8bit_adam'
114
  if train_text_encoder:
115
- command += f' --train_text_encoder --learning_rate_text={learning_rate_text} --color_jitter'
116
 
117
  with open(self.output_dir / 'train.sh', 'w') as f:
118
  command_s = ' '.join(command.split())
 
112
  if use_8bit_adam:
113
  command += ' --use_8bit_adam'
114
  if train_text_encoder:
115
+ command += f' --train_text_encoder --learning_rate_text={learning_rate_text}'
116
 
117
  with open(self.output_dir / 'train.sh', 'w') as f:
118
  command_s = ' '.join(command.split())