Spaces:
Runtime error
Runtime error
ShaoTengLiu
commited on
Commit
·
f56167c
1
Parent(s):
1765eca
debug
Browse files- 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:
|