Spaces:
Build error
Build error
Commit
·
9110980
1
Parent(s):
6a97866
Update app.py
Browse files
app.py
CHANGED
|
@@ -369,6 +369,11 @@ Sample pictures of:
|
|
| 369 |
return [gr.update(visible=True, value=f"Successfully uploaded your model. Access it [here](https://huggingface.co/{model_id})"), gr.update(visible=True, value=["diffusers_model.tar", "model.ckpt"])]
|
| 370 |
|
| 371 |
def convert_to_ckpt():
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 372 |
convert("output_model", "model.ckpt")
|
| 373 |
return gr.update(visible=True, value=["diffusers_model.tar", "model.ckpt"])
|
| 374 |
|
|
|
|
| 369 |
return [gr.update(visible=True, value=f"Successfully uploaded your model. Access it [here](https://huggingface.co/{model_id})"), gr.update(visible=True, value=["diffusers_model.tar", "model.ckpt"])]
|
| 370 |
|
| 371 |
def convert_to_ckpt():
|
| 372 |
+
if 'pipe' in globals():
|
| 373 |
+
global pipe, pipe_is_set
|
| 374 |
+
del pipe
|
| 375 |
+
pipe_is_set = False
|
| 376 |
+
gc.collect()
|
| 377 |
convert("output_model", "model.ckpt")
|
| 378 |
return gr.update(visible=True, value=["diffusers_model.tar", "model.ckpt"])
|
| 379 |
|