Commit
•
126bc9f
1
Parent(s):
0feb48e
Update app.py
Browse files
app.py
CHANGED
@@ -159,10 +159,10 @@ def train(*inputs):
|
|
159 |
max_train_steps=Training_Steps,
|
160 |
)
|
161 |
run_training(args_general)
|
|
|
162 |
convert("output_model", "model.ckpt")
|
163 |
shutil.rmtree('instance_images')
|
164 |
shutil.make_archive("diffusers_model", 'zip', "output_model")
|
165 |
-
#shutil.rmtree("output_model")
|
166 |
torch.cuda.empty_cache()
|
167 |
return [gr.update(visible=True, value=["diffusers_model.zip", "model.ckpt"]), gr.update(visible=True), gr.update(visible=True)]
|
168 |
|
|
|
159 |
max_train_steps=Training_Steps,
|
160 |
)
|
161 |
run_training(args_general)
|
162 |
+
torch.cuda.empty_cache()
|
163 |
convert("output_model", "model.ckpt")
|
164 |
shutil.rmtree('instance_images')
|
165 |
shutil.make_archive("diffusers_model", 'zip', "output_model")
|
|
|
166 |
torch.cuda.empty_cache()
|
167 |
return [gr.update(visible=True, value=["diffusers_model.zip", "model.ckpt"]), gr.update(visible=True), gr.update(visible=True)]
|
168 |
|