Spaces:
Runtime error
Runtime error
RohitGandikota
commited on
Commit
Β·
2fc9d25
1
Parent(s):
39c1245
fixing training
Browse files
app.py
CHANGED
@@ -227,13 +227,11 @@ class Demo:
|
|
227 |
)
|
228 |
|
229 |
def train(self, target_concept,positive_prompt, negative_prompt, rank, iterations_input, lr_input, train_method, neg_guidance, iterations, lr, attributes_input, is_person, pbar = gr.Progress(track_tqdm=True)):
|
230 |
-
if
|
231 |
-
attributes_input = None
|
232 |
-
if '...' in target_concept:
|
233 |
target_concept = ''
|
234 |
-
if
|
235 |
positive_prompt = ''
|
236 |
-
if
|
237 |
negative_prompt = ''
|
238 |
|
239 |
|
|
|
227 |
)
|
228 |
|
229 |
def train(self, target_concept,positive_prompt, negative_prompt, rank, iterations_input, lr_input, train_method, neg_guidance, iterations, lr, attributes_input, is_person, pbar = gr.Progress(track_tqdm=True)):
|
230 |
+
if target_concept is None:
|
|
|
|
|
231 |
target_concept = ''
|
232 |
+
if positive_prompt is None:
|
233 |
positive_prompt = ''
|
234 |
+
if negative_prompt is None:
|
235 |
negative_prompt = ''
|
236 |
|
237 |
|