update

app.py

@@ -5,8 +5,8 @@ import os
 # print(f"pip3 uninstall gradio")
 # results = os.popen(f"pip3 uninstall gradio -y")
 # print(results.readlines())
-results = os.popen(f"ls -lR  ~/.cache/huggingface/hub")
-print("step0:", results.readlines())
+# results = os.popen(f"ls -lR  ~/.cache/huggingface/hub")
+# print("step0:", results.readlines())
 results = os.popen(f"pip3 install -r requirements.txt -i https://pypi.org/simple")
 print("step1:", results.readlines())
 results = os.popen(f"pip3 uninstall scepter -y")
@@ -15,8 +15,8 @@ results = os.popen(f"pip3 install scepter -i https://pypi.org/simple")
 print("step3:", results.readlines())
 results = os.popen(f"pip3 list")
 print("step4:", results.readlines())
-results = os.popen(f"sudo chmod 777 /data")
-print("step5:", results.readlines())
+#results = os.popen(f"sudo chmod 777 /data")
+#print("step5:", results.readlines())
 print("finish install")
 
 import random
@@ -27,6 +27,29 @@ from scepter.modules.utils.config import Config
 from scepter.modules.utils.file_system import FS
 from scepter.modules.utils.logger import get_logger, init_logger
 
+def link_model():
+    import os
+    ms_root_dir = "cache/cache_data/"
+    hf_root_dir = "/home/user/.cache/huggingface/hub/"
+    # hf_root_dir = "/root/.cache/huggingface/hub/"
+    model_map = {
+        "95c8c1dcca842facbdcb9606c314f7ae/AI-ModelScope/stable-diffusion-v1-5": "models--runwayml--stable-diffusion-v1-5/snapshots/1d0c4ebf6ff58a5caecab40fa1406526bca4b5b9",
+        "601abc72ccca722467955bb9c36da4b3/AI-ModelScope/stable-diffusion-2-1": "models--stabilityai--stable-diffusion-2-1/snapshots/5cae40e6a2745ae2b01ad92ae5043f95f23644d6",
+        "76ee531623cf180d9c98a6fbf5fa0c45.0/AI-ModelScope/stable-diffusion-xl-base-1.0": "models--stabilityai--stable-diffusion-xl-base-1.0/snapshots/462165984030d82259a11f4367a4eed129e94a7b",
+        "37fb512464d701519e3be1d14c97f8c0/AI-ModelScope/clip-vit-large-patch14": "models--openai--clip-vit-large-patch14/snapshots/32bd64288804d66eefd0ccbe215aa642df71cc41",
+        "c2412c56551dfb7baa71c84978ee22eb/iic/scepter": "models--scepter-studio--scepter/snapshots/d78fd86694550622e1d256865fe444282fcccb19",
+        "d5bec966515f3501f44f4844c8f8e5cb/iic/scepter_scedit": "models--scepter-studio--scepter_scedit/snapshots/c8905ae6dda1ebafd2db20261ebe98cf0d9a2cd1",
+        "578c6574e9925c75f1ae50888bf65ccb/damo/LARGEN": "models--scepter-studio--LAR-Gen/snapshots/7ab4b7d934d60ed80600f2971fa0c4004ba17b4d",
+        "b2e04aa8710d90c9486f14465c543e83/damo/stylebooth": "models--scepter-studio--stylebooth/snapshots/c993df593ed451c137118b5426a1dcab29e38d5c",
+        "7168a5d7e0eb13ef4facee9a8d2da34c/cubeai/blip-image-captioning-base": "models--Salesforce--blip-image-captioning-base/snapshots/89b09ea1789f7addf2f6d6f0dfc4ce10ab58ef84"
+    }
+    for ms_name, hf_name in model_map.items():
+        ms_path =os.path.join(ms_root_dir, ms_name)
+        hf_path = os.path.join(hf_root_dir, hf_name)
+        cmd = f"mkdir -p {ms_path} && ln -s {hf_path}/* {ms_path}"
+        os.system(cmd)
+
+link_model()
 
 def prepare(config):
     if 'FILE_SYSTEM' in config:
@@ -109,4 +132,4 @@ with gr.Blocks() as demo:
         for interface, label, ifid in interfaces:
             interface.set_callbacks(tab_manager)
 
-demo.queue(status_update_rate=1, api_open=False).launch(share=False, show_error=True, enable_queue=True)
+demo.queue(status_update_rate=1, api_open=False).launch(share=False, show_error=True, enable_queue=True)