test

app.py

@@ -5,7 +5,7 @@ import spaces
 from huggingface_hub import hf_hub_download
 from utils import utils, tools, preprocess
 
-BASE_MODEL_PATH = "stablediffusionapi/neta-art-xl-v2"
+# 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"
@@ -23,6 +23,11 @@ def ui():
     #     filename='AAM_XL_Anime_Mix.safetensors',
     #     cache_dir=CACHE_DIR,
     # )
+    model_file = hf_hub_download(
+        repo_id='Eugeoter/artiwaifu-diffusion-1.0',
+        filename='artiwaifu-diffusion-v1.safetensors',
+        cache_dir=CACHE_DIR,
+    )
     unet_file = hf_hub_download(
         repo_id=REPO_ID,
         filename=UNET_FILENAME,
@@ -35,7 +40,7 @@ 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,
@@ -43,8 +48,7 @@ def ui():
         load_weight_increasement=True,
         device=device,
         hf_cache_dir=CACHE_DIR,
-        use_safetensors=False,
-        variant=None,
+        use_safetensors=True,
     )
 
     preprocessors = ['canny']