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"
@@ -18,11 +18,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='Lykon/AAM_XL_AnimeMix',
-        filename='AAM_XL_Anime_Mix.safetensors',
-        cache_dir=CACHE_DIR,
-    )
+    # model_file = hf_hub_download(
+    #     repo_id='Lykon/AAM_XL_AnimeMix',
+    #     filename='AAM_XL_Anime_Mix.safetensors',
+    #     cache_dir=CACHE_DIR,
+    # )
     unet_file = hf_hub_download(
         repo_id=REPO_ID,
         filename=UNET_FILENAME,
@@ -35,7 +35,7 @@ def ui():
     )
 
     pipeline = tools.get_pipeline(
-        pretrained_model_name_or_path=model_file,
+        pretrained_model_name_or_path=BASE_MODEL_PATH,
         unet_model_name_or_path=unet_file,
         controlnet_model_name_or_path=controlnet_file,
         vae_model_name_or_path=VAE_PATH,
@@ -43,7 +43,8 @@ def ui():
         load_weight_increasement=True,
         device=device,
         hf_cache_dir=CACHE_DIR,
-        use_safetensors=True,
+        use_safetensors=False,
+        variant=None,
     )
 
     preprocessors = ['canny']