yonishafir commited on
Commit
118d6a9
·
1 Parent(s): 44ad21a
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -173,7 +173,8 @@ from huggingface_hub import hf_hub_download
173
  hf_hub_download(repo_id="briaai/ID_preservation_2.3", filename="controlnet/config.json", local_dir="./checkpoints")
174
  hf_hub_download(repo_id="briaai/ID_preservation_2.3", filename="controlnet/diffusion_pytorch_model.safetensors", local_dir="./checkpoints")
175
  hf_hub_download(repo_id="briaai/ID_preservation_2.3", filename="ip-adapter.bin", local_dir="./checkpoints")
176
- hf_hub_download(repo_id="briaai/ID_preservation_2.3", filename="image_encoder/*", local_dir="./checkpoints")
 
177
 
178
 
179
 
@@ -197,7 +198,7 @@ controlnet = [controlnet_lnmks, controlnet_canny]
197
  device = "cuda" if torch.cuda.is_available() else "cpu"
198
 
199
  image_encoder = CLIPVisionModelWithProjection.from_pretrained(
200
- f"./checkpoints//image_encoder",
201
  torch_dtype=torch.float16,
202
  )
203
 
 
173
  hf_hub_download(repo_id="briaai/ID_preservation_2.3", filename="controlnet/config.json", local_dir="./checkpoints")
174
  hf_hub_download(repo_id="briaai/ID_preservation_2.3", filename="controlnet/diffusion_pytorch_model.safetensors", local_dir="./checkpoints")
175
  hf_hub_download(repo_id="briaai/ID_preservation_2.3", filename="ip-adapter.bin", local_dir="./checkpoints")
176
+ hf_hub_download(repo_id="briaai/ID_preservation_2.3", filename="image_encoder/pytorch_model.bin", local_dir="./checkpoints")
177
+ hf_hub_download(repo_id="briaai/ID_preservation_2.3", filename="image_encoder/config.json", local_dir="./checkpoints")
178
 
179
 
180
 
 
198
  device = "cuda" if torch.cuda.is_available() else "cpu"
199
 
200
  image_encoder = CLIPVisionModelWithProjection.from_pretrained(
201
+ f"./checkpoints/image_encoder",
202
  torch_dtype=torch.float16,
203
  )
204