briaai
/

BRIA-2.3-ID_Preservation

legal liability

ID preservation Adapter

Model card Files Files and versions Community

yonishafir commited on Jan 12

Commit

5a1c952

·

verified ·

1 Parent(s): 52141a6

Update README.md

Files changed (1) hide show

README.md +12 -9

README.md CHANGED Viewed

@@ -115,17 +115,20 @@ from pipeline_bria_id_preservation import BriaIDPreservationDiffusionPipeline, d
 # =================  Prepare and download models and checkpoints =================
 # Download face encoder
-snapshot_download(
-    "fal/AuraFace-v1",
-    local_dir="./models/auraface",
-)
 # download checkpoints
-hf_hub_download(repo_id="briaai/ID_preservation_2.3_auraFaceEnc", filename="checkpoint_105000/controlnet/config.json", local_dir="./checkpoints")
-hf_hub_download(repo_id="briaai/ID_preservation_2.3_auraFaceEnc", filename="checkpoint_105000/controlnet/diffusion_pytorch_model.safetensors", local_dir="./checkpoints")
-hf_hub_download(repo_id="briaai/ID_preservation_2.3_auraFaceEnc", filename="checkpoint_105000/ip-adapter.bin", local_dir="./checkpoints")
-hf_hub_download(repo_id="briaai/ID_preservation_2.3_auraFaceEnc", filename="image_encoder/pytorch_model.bin", local_dir="./checkpoints")
-hf_hub_download(repo_id="briaai/ID_preservation_2.3_auraFaceEnc", filename="image_encoder/config.json", local_dir="./checkpoints")
 # Util functions

 # =================  Prepare and download models and checkpoints =================
 # Download face encoder
+snapshot_download("fal/AuraFace-v1", local_dir="./models/auraface")
 # download checkpoints
+hf_hub_download(repo_id="briaai/BRIA-2.3-ID_Preservation", filename="checkpoint_105000/controlnet/config.json", local_dir="./checkpoints")
+hf_hub_download(repo_id="briaai/BRIA-2.3-ID_Preservation", filename="checkpoint_105000/controlnet/diffusion_pytorch_model.safetensors", local_dir="./checkpoints")
+hf_hub_download(repo_id="briaai/BRIA-2.3-ID_Preservation", filename="checkpoint_105000/ip-adapter.bin", local_dir="./checkpoints")
+hf_hub_download(repo_id="briaai/BRIA-2.3-ID_Preservation", filename="image_encoder/pytorch_model.bin", local_dir="./checkpoints")
+hf_hub_download(repo_id="briaai/BRIA-2.3-ID_Preservation", filename="image_encoder/config.json", local_dir="./checkpoints")
+# download needed files
+hf_hub_download(repo_id="briaai/BRIA-2.3-ID_Preservation", filename="pipeline_bria_id_preservation.py", local_dir=".")
+hf_hub_download(repo_id="briaai/BRIA-2.3-ID_Preservation", filename="ip_adapter/attention_processor.py", local_dir="./ip_adapter")
+hf_hub_download(repo_id="briaai/BRIA-2.3-ID_Preservation", filename="ip_adapter/resampler.py", local_dir="./ip_adapter")
+hf_hub_download(repo_id="briaai/BRIA-2.3-ID_Preservation", filename="ip_adapter/utils.py", local_dir="./ip_adapter")
 # Util functions