RohitGandikota commited on
Commit
67acccc
β€’
1 Parent(s): ccd7636

enabling training for xattn

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = max(int(iterations_input),1000)
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)