multimodalart HF staff commited on
Commit
36aeb40
1 Parent(s): e4068c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -76,7 +76,7 @@ def train(*inputs):
76
  Train_text_encoder_for = int(inputs[-2])
77
  else:
78
  Training_Steps = file_counter*200
79
- if(type_of_thing == "person"):
80
  class_data_dir = "mix"
81
  Train_text_encoder_for=100
82
  args_txt_encoder = argparse.Namespace(
@@ -126,7 +126,7 @@ def train(*inputs):
126
  )
127
  run_training(args_txt_encoder)
128
  run_training(args_unet)
129
- elif(type_of_thing == "object" or type_of_thing == "style"):
130
  if(type_of_thing == "object"):
131
  Train_text_encoder_for=30
132
  elif(type_of_thing == "style"):
 
76
  Train_text_encoder_for = int(inputs[-2])
77
  else:
78
  Training_Steps = file_counter*200
79
+ if(type_of_thing == "person"):
80
  class_data_dir = "mix"
81
  Train_text_encoder_for=100
82
  args_txt_encoder = argparse.Namespace(
 
126
  )
127
  run_training(args_txt_encoder)
128
  run_training(args_unet)
129
+ elif(type_of_thing == "object" or type_of_thing == "style"):
130
  if(type_of_thing == "object"):
131
  Train_text_encoder_for=30
132
  elif(type_of_thing == "style"):