better model loading

README.md

@@ -7,8 +7,6 @@ sdk: gradio
 sdk_version: 4.21.0
 app_file: app.py
 pinned: false
-preload_from_hub:
-  - lllyasviel/fav_models fav/juggernautXL_v8Rundiffusion.safetensors
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -1,6 +1,8 @@
 import sys
 import os
 import torch
+from pathlib import Path
+from huggingface_hub import hf_hub_download
 from PIL import Image, ImageSequence, ImageOps
 from typing import List
 import numpy as np
@@ -27,6 +29,19 @@ from ComfyUI.custom_nodes.layerdiffuse.layered_diffusion import (
 import gradio as gr
 
 
+MODEL_PATH = hf_hub_download(
+    repo_id="lllyasviel/fav_models",
+    subfolder="fav",
+    filename="juggernautXL_v8Rundiffusion.safetensors",
+)
+try:
+    os.symlink(
+        MODEL_PATH,
+        Path("./ComfyUI/models/checkpoints/juggernautXL_v8Rundiffusion.safetensors"),
+    )
+except FileExistsError:
+    pass
+
 with torch.inference_mode():
     ckpt_load_checkpoint = CheckpointLoaderSimple().load_checkpoint
     ckpt = ckpt_load_checkpoint(ckpt_name="juggernautXL_v8Rundiffusion.safetensors")