Eugeoter commited on
Commit
0b60317
·
1 Parent(s): 6c9063a
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -5,7 +5,7 @@ import spaces
5
  from huggingface_hub import hf_hub_download
6
  from utils import utils, tools, preprocess
7
 
8
- BASE_MODEL_PATH = "stablediffusionapi/neta-art-xl-v2"
9
  VAE_PATH = "madebyollin/sdxl-vae-fp16-fix"
10
  REPO_ID = "Pbihao/ControlNeXt"
11
  UNET_FILENAME = "ControlAny-SDXL/anime_canny/unet.safetensors"
@@ -23,6 +23,11 @@ def ui():
23
  # filename='AAM_XL_Anime_Mix.safetensors',
24
  # cache_dir=CACHE_DIR,
25
  # )
 
 
 
 
 
26
  unet_file = hf_hub_download(
27
  repo_id=REPO_ID,
28
  filename=UNET_FILENAME,
@@ -35,7 +40,7 @@ def ui():
35
  )
36
 
37
  pipeline = tools.get_pipeline(
38
- pretrained_model_name_or_path=BASE_MODEL_PATH,
39
  unet_model_name_or_path=unet_file,
40
  controlnet_model_name_or_path=controlnet_file,
41
  vae_model_name_or_path=VAE_PATH,
@@ -43,8 +48,7 @@ def ui():
43
  load_weight_increasement=True,
44
  device=device,
45
  hf_cache_dir=CACHE_DIR,
46
- use_safetensors=False,
47
- variant=None,
48
  )
49
 
50
  preprocessors = ['canny']
 
5
  from huggingface_hub import hf_hub_download
6
  from utils import utils, tools, preprocess
7
 
8
+ # BASE_MODEL_PATH = "stablediffusionapi/neta-art-xl-v2"
9
  VAE_PATH = "madebyollin/sdxl-vae-fp16-fix"
10
  REPO_ID = "Pbihao/ControlNeXt"
11
  UNET_FILENAME = "ControlAny-SDXL/anime_canny/unet.safetensors"
 
23
  # filename='AAM_XL_Anime_Mix.safetensors',
24
  # cache_dir=CACHE_DIR,
25
  # )
26
+ model_file = hf_hub_download(
27
+ repo_id='Eugeoter/artiwaifu-diffusion-1.0',
28
+ filename='artiwaifu-diffusion-v1.safetensors',
29
+ cache_dir=CACHE_DIR,
30
+ )
31
  unet_file = hf_hub_download(
32
  repo_id=REPO_ID,
33
  filename=UNET_FILENAME,
 
40
  )
41
 
42
  pipeline = tools.get_pipeline(
43
+ pretrained_model_name_or_path=model_file,
44
  unet_model_name_or_path=unet_file,
45
  controlnet_model_name_or_path=controlnet_file,
46
  vae_model_name_or_path=VAE_PATH,
 
48
  load_weight_increasement=True,
49
  device=device,
50
  hf_cache_dir=CACHE_DIR,
51
+ use_safetensors=True,
 
52
  )
53
 
54
  preprocessors = ['canny']