RohitGandikota commited on
Commit
2fc9d25
Β·
1 Parent(s): 39c1245

fixing training

Browse files
Files changed (1) hide show
  1. app.py +3 -5
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 '' in attributes_input:
231
- attributes_input = None
232
- if '...' in target_concept:
233
  target_concept = ''
234
- if '...' in positive_prompt:
235
  positive_prompt = ''
236
- if '...' in negative_prompt:
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