Commit
•
36aeb40
1
Parent(s):
e4068c9
Update app.py
Browse files
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 |
-
|
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 |
-
|
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"):
|