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