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