刘虹雨 commited on
Commit
4c94a4e
·
1 Parent(s): e52700c

update code

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -323,6 +323,10 @@ def launch_pretrained():
323
  local_dir="./pretrained_model/sd21",
324
  local_dir_use_symlinks=False
325
  )
 
 
 
 
326
 
327
  # 下载 CrucibleAI/ControlNetMediaPipeFace 的所有文件
328
  snapshot_download(
 
323
  local_dir="./pretrained_model/sd21",
324
  local_dir_use_symlinks=False
325
  )
326
+ logging.info("delete models.")
327
+
328
+ os.remove('./pretrained_model/sd21/v2-1_512-ema-pruned.ckpt')
329
+ os.remove('./pretrained_model/sd21/v2-1_512-nonema-pruned.ckpt')
330
 
331
  # 下载 CrucibleAI/ControlNetMediaPipeFace 的所有文件
332
  snapshot_download(