Spaces:
Build error
Build error
Update trainer.py
Browse files- 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}
|
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())
|