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 |
|