cocktailpeanut commited on
Commit
8c19a23
·
1 Parent(s): bd9d73c
Files changed (1) hide show
  1. app.py +10 -10
app.py CHANGED
@@ -88,33 +88,33 @@ sdxl_loras_raw = [item for item in sdxl_loras_raw if item.get("new") != True]
88
  hf_hub_download(
89
  repo_id="InstantX/InstantID",
90
  filename="ControlNetModel/config.json",
91
- # local_dir="data/checkpoints",
92
  )
93
  hf_hub_download(
94
  repo_id="InstantX/InstantID",
95
  filename="ControlNetModel/diffusion_pytorch_model.safetensors",
96
- # local_dir="data/checkpoints",
97
  )
98
  hf_hub_download(
99
  repo_id="InstantX/InstantID", filename="ip-adapter.bin",
100
- # local_dir="data/checkpoints"
101
  )
102
  hf_hub_download(
103
  repo_id="latent-consistency/lcm-lora-sdxl",
104
  filename="pytorch_lora_weights.safetensors",
105
- # local_dir="data/checkpoints",
106
  )
107
  # download antelopev2
108
- if not os.path.exists("/data/antelopev2.zip"):
109
- gdown.download(url="https://drive.google.com/file/d/18wEUfMNohBJ4K3Ly5wpTejPfDzp-8fI8/view?usp=sharing", output="/data/", quiet=False, fuzzy=True)
110
- os.system("unzip /data/antelopev2.zip -d /data/models/")
111
 
112
- app = FaceAnalysis(name='antelopev2', root='/data', providers=['CPUExecutionProvider'])
113
  app.prepare(ctx_id=0, det_size=(640, 640))
114
 
115
  # prepare models under ./checkpoints
116
- face_adapter = f'/data/checkpoints/ip-adapter.bin'
117
- controlnet_path = f'/data/checkpoints/ControlNetModel'
118
 
119
  # load IdentityNet
120
  st = time.time()
 
88
  hf_hub_download(
89
  repo_id="InstantX/InstantID",
90
  filename="ControlNetModel/config.json",
91
+ local_dir="data/checkpoints",
92
  )
93
  hf_hub_download(
94
  repo_id="InstantX/InstantID",
95
  filename="ControlNetModel/diffusion_pytorch_model.safetensors",
96
+ local_dir="data/checkpoints",
97
  )
98
  hf_hub_download(
99
  repo_id="InstantX/InstantID", filename="ip-adapter.bin",
100
+ local_dir="data/checkpoints"
101
  )
102
  hf_hub_download(
103
  repo_id="latent-consistency/lcm-lora-sdxl",
104
  filename="pytorch_lora_weights.safetensors",
105
+ local_dir="data/checkpoints",
106
  )
107
  # download antelopev2
108
+ if not os.path.exists("data/antelopev2.zip"):
109
+ gdown.download(url="https://drive.google.com/file/d/18wEUfMNohBJ4K3Ly5wpTejPfDzp-8fI8/view?usp=sharing", output="data/", quiet=False, fuzzy=True)
110
+ os.system("unzip data/antelopev2.zip -d data/models/")
111
 
112
+ app = FaceAnalysis(name='antelopev2', root='data', providers=['CPUExecutionProvider'])
113
  app.prepare(ctx_id=0, det_size=(640, 640))
114
 
115
  # prepare models under ./checkpoints
116
+ face_adapter = f'data/checkpoints/ip-adapter.bin'
117
+ controlnet_path = f'data/checkpoints/ControlNetModel'
118
 
119
  # load IdentityNet
120
  st = time.time()