ShaoTengLiu commited on
Commit
f56167c
·
1 Parent(s): 1765eca
Files changed (1) hide show
  1. trainer.py +1 -1
trainer.py CHANGED
@@ -135,7 +135,7 @@ class Trainer:
135
  config.use_8bit_adam = use_8bit_adam
136
  config.prompts = [training_prompt, validation_prompt]
137
  config.blend_word = [blend_word_1, blend_word_2]
138
- config.eq_params = {"words":eq_params_1, "values":int(eq_params_2)}
139
  if len(validation_prompt) == len(training_prompt):
140
  config.is_word_swap = True
141
  else:
 
135
  config.use_8bit_adam = use_8bit_adam
136
  config.prompts = [training_prompt, validation_prompt]
137
  config.blend_word = [blend_word_1, blend_word_2]
138
+ config.eq_params = {"words":[eq_params_1], "values":[int(eq_params_2)]}
139
  if len(validation_prompt) == len(training_prompt):
140
  config.is_word_swap = True
141
  else: