add preload and jemalloc, libiomp

README.md

@@ -8,6 +8,10 @@ sdk_version: 4.24.0
 app_file: app.py
 pinned: false
 license: cc-by-nc-4.0
+preload_from_hub:
+    - SG161222/Realistic_Vision_V4.0_noVAE
+    - stabilityai/sd-vae-ft-mse
+    - runwayml/stable-diffusion-v1-5 scheduler/scheduler_config.json
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -1,4 +1,18 @@
 # adapted from https://huggingface.co/spaces/HumanAIGC/OutfitAnyone/blob/main/app.py
+import os
+from os.path import join as opj
+
+LD_PRELOAD = os.getenv("LD_PRELOAD")
+os.environ["LD_PRELOAD"] = f"{LD_PRELOAD}:/usr/lib/x86_64-linux-gnu/libjemalloc.so"
+os.environ["MALLOC_CONF"] = "oversize_threshold:1,background_thread:true,metadata_thp:auto,dirty_decay_ms: 60000,muzzy_decay_ms:60000"
+
+LD_PRELOAD= os.getenv("LD_PRELOAD")
+os.environ["LD_PRELOAD"] = f"{LD_PRELOAD}:/usr/lib/x86_64-linux-gnu/libjemalloc.so:/usr/lib/x86_64-linux-gnu/libiomp5.so"
+os.environ["OMP_NUM_THREADS"] ==32
+
+token = os.getenv("ACCESS_TOKEN")
+os.system(f"python -m pip install git+https://{token}@github.com/logn-2024/StableGarment.git")
+
 import torch
 import spaces
 import gradio as gr
@@ -14,12 +28,6 @@ from diffusers import StableDiffusionPipeline
 from diffusers.loaders import LoraLoaderMixin
 import intel_extension_for_pytorch as ipex
 
-import os
-from os.path import join as opj
-
-token = os.getenv("ACCESS_TOKEN")
-os.system(f"python -m pip install git+https://{token}@github.com/logn-2024/StableGarment.git")
-
 from stablegarment.models import GarmentEncoderModel,ControlNetModel
 from stablegarment.piplines import StableGarmentPipeline,StableGarmentControlNetPipeline