updated

app.py

@@ -5,7 +5,6 @@ import spaces
 from huggingface_hub import hf_hub_download
 from utils import utils, tools, preprocess
 
-BASE_MODEL_PATH = "stablediffusionapi/neta-art-xl-v2"
 VAE_PATH = "madebyollin/sdxl-vae-fp16-fix"
 REPO_ID = "Pbihao/ControlNeXt"
 UNET_FILENAME = "ControlAny-SDXL/anime_canny/unet.safetensors"
@@ -18,6 +17,11 @@ DEFAULT_NEGATIVE_PROMPT = "worst quality, abstract, clumsy pose, deformed hand,
 
 def ui():
     device = "cuda" if torch.cuda.is_available() else "cpu"
+    model_file = hf_hub_download(
+        repo_id='neta-art/neta-xl-2.0',
+        filename='neta-xl-v2.fp16.safetensors',
+        cache_dir=CACHE_DIR,
+    )
     unet_file = hf_hub_download(
         repo_id=REPO_ID,
         filename=UNET_FILENAME,
@@ -30,15 +34,14 @@ def ui():
     )
 
     pipeline = tools.get_pipeline(
-        pretrained_model_name_or_path=BASE_MODEL_PATH,
+        pretrained_model_name_or_path=model_file,
         unet_model_name_or_path=unet_file,
         controlnet_model_name_or_path=controlnet_file,
         vae_model_name_or_path=VAE_PATH,
-
         load_weight_increasement=True,
         device=device,
         hf_cache_dir=CACHE_DIR,
-        use_safetensors=False,
+        use_safetensors=True,
     )
 
     preprocessors = ['canny']