multimodalart HF staff commited on
Commit
9110980
1 Parent(s): 6a97866

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
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