Spaces:
Runtime error
Runtime error
RohitGandikota
commited on
Commit
Β·
67acccc
1
Parent(s):
ccd7636
enabling training for xattn
Browse files
app.py
CHANGED
|
@@ -235,7 +235,7 @@ class Demo:
|
|
| 235 |
)
|
| 236 |
|
| 237 |
def train(self, target_concept,positive_prompt, negative_prompt, rank, iterations_input, lr_input, attributes_input, is_person, pbar = gr.Progress(track_tqdm=True)):
|
| 238 |
-
iterations_input =
|
| 239 |
if attributes_input == '':
|
| 240 |
attributes_input = None
|
| 241 |
print(target_concept, positive_prompt, negative_prompt, attributes_input, is_person)
|
|
|
|
| 235 |
)
|
| 236 |
|
| 237 |
def train(self, target_concept,positive_prompt, negative_prompt, rank, iterations_input, lr_input, attributes_input, is_person, pbar = gr.Progress(track_tqdm=True)):
|
| 238 |
+
iterations_input = min(int(iterations_input),1000)
|
| 239 |
if attributes_input == '':
|
| 240 |
attributes_input = None
|
| 241 |
print(target_concept, positive_prompt, negative_prompt, attributes_input, is_person)
|