From c34de4d6b0879c75fce909f1a2ef1730a2a90cf7 Mon Sep 17 00:00:00 2001
From: Yuwei Guo <guoyuwei@pjlab.org.cn>
Date: Thu, 20 Jul 2023 14:37:31 +0800
Subject: [PATCH] update

---
 app.py | 41 +++++++++++++++++++++--------------------
 1 file changed, 21 insertions(+), 20 deletions(-)

diff --git a/app.py b/app.py
index e2ff368..1dd62db 100644
--- a/app.py
+++ b/app.py
@@ -65,7 +65,7 @@ examples = [
     ],
     # 5-RealisticVision
     [
-        "realisticVisionV20_v20.safetensors", 
+        "realisticVisionV40_v20Novae.safetensors", 
         "mm_sd_v15.ckpt", 
         "photo of coastline, rocks, storm weather, wind, waves, lightning, 8k uhd, dslr, soft lighting, high quality, film grain, Fujifilm XT3",
         "blur, haze, deformed iris, deformed pupils, semi-realistic, cgi, 3d, render, sketch, cartoon, drawing, anime, mutated hands and fingers, deformed, distorted, disfigured, poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, disconnected limbs, mutation, mutated, ugly, disgusting, amputation",
@@ -74,27 +74,28 @@ examples = [
 ]
 
 # clean unrelated ckpts
-# ckpts = [
-#     "realisticVisionV20_v20.safetensors",
-#     "majicmixRealistic_v5Preview.safetensors",
-#     "rcnzCartoon3d_v10.safetensors",
-#     "lyriel_v16.safetensors",
-#     "toonyou_beta3.safetensors"
-# ]
+ckpts = [
+    "realisticVisionV40_v20Novae.safetensors",
+    "majicmixRealistic_v5Preview.safetensors",
+    "rcnzCartoon3d_v10.safetensors",
+    "lyriel_v16.safetensors",
+    "toonyou_beta3.safetensors"
+]
 
-# for path in glob(os.path.join("models", "DreamBooth_LoRA", "*.safetensors")):
-#     for ckpt in ckpts:
-#         if path.endswith(ckpt): break
-#     else:
-#         print(f"### Cleaning {path} ...")
-#         os.system(f"rm -rf {path}")
-os.system(f"rm -rf {os.path.join('models', 'DreamBooth_LoRA', '*.safetensors')}")
+for path in glob(os.path.join("models", "DreamBooth_LoRA", "*.safetensors")):
+    for ckpt in ckpts:
+        if path.endswith(ckpt): break
+    else:
+        print(f"### Cleaning {path} ...")
+        os.system(f"rm -rf {path}")
 
-os.system(f"bash download_bashscripts/1-ToonYou.sh")
-os.system(f"bash download_bashscripts/2-Lyriel.sh")
-os.system(f"bash download_bashscripts/3-RcnzCartoon.sh")
-os.system(f"bash download_bashscripts/4-MajicMix.sh")
-os.system(f"bash download_bashscripts/5-RealisticVision.sh")
+# os.system(f"rm -rf {os.path.join('models', 'DreamBooth_LoRA', '*.safetensors')}")
+
+# os.system(f"bash download_bashscripts/1-ToonYou.sh")
+# os.system(f"bash download_bashscripts/2-Lyriel.sh")
+# os.system(f"bash download_bashscripts/3-RcnzCartoon.sh")
+# os.system(f"bash download_bashscripts/4-MajicMix.sh")
+# os.system(f"bash download_bashscripts/5-RealisticVision.sh")
 
 # clean Grdio cache
 print(f"### Cleaning cached examples ...")