PuLID (NeurIPS 2024)

:open_book: PuLID: Pure and Lightning ID Customization via Contrastive Alignment

Zinan Guo, Yanze Wu✝, Zhuowei Chen, Lang Chen, Peng Zhang, Qian He
(*Equal Contribution, ✝Corresponding Author)
ByteDance Inc

:triangular_flag_on_post: Updates

2024.09.26: 🎉 PuLID accepted by NeurIPS 2024
2024.09.12: 💥 We're thrilled to announce the release of the PuLID-FLUX-v0.9.0 model. Enjoy exploring its capabilities! 😊 Learn more about this model
2024.05.23: share the preview of our upcoming v1.1 model, please stay tuned
2024.05.01: release v1 codes&models, also the 🤗HuggingFace Demo
2024.04.25: release arXiv paper.

:soon: update plan

[ ] release PuLID-FLUX-v0.9.1 model in 2024.10
[ ] release PuLID v1.1 (for SDXL) model in 2024.10

PuLID for FLUX

Please check the doc and demo of PuLID-FLUX here.

We will actively update and maintain this repository in the near future, so please stay tuned.

updates

[x] Local gradio demo is ready now
[x] Online HuggingFace demo is ready now
[x] We have optimized the codes to support consumer-grade GPUS, and now PuLID-FLUX can run on a 16GB graphic card. Check the details here
[x] (Community Implementation) Online Replicate demo is ready now
[x] Local gradio demo supports 12GB graphic card now

Below results are generated with PuLID-FLUX. pulid_flux_results

Examples

Images generated with our PuLID examples Applications

https://github.com/ToTheBeginning/PuLID/assets/11482921/9bdd0c8a-99e8-4eab-ab9e-39bf796cc6b8

:wrench: Dependencies and Installation

Python >= 3.9 (Recommend to use Anaconda or Miniconda)

PyTorch >= 2.0 if you don't need flux-dev-fp8, otherwise PyTorch >= 2.4.1

# clone PuLID repo
git clone https://github.com/ToTheBeginning/PuLID.git
cd PuLID
# create conda env
conda create --name pulid python=3.10
# activate env
conda activate pulid
# Install dependent packages
# 1. if you don't need flux-fp8, e.g., you are using xl or flux-bf16, install the following requirements.txt
pip install -r requirements.txt
# 2. if you need flux-fp8 (to put flux on consumer-grade gpu), install the following requirements_fp8.txt
pip install -r requirements_fp8.txt

:zap: Quick Inference

Local Gradio Demo

python app.py

Online HuggingFace Demo

Thanks for the GPU grant from HuggingFace team, you can try PuLID HF demo in https://huggingface.co/spaces/yanze/PuLID

:paperclip: Related Resources

Following are some third-party implementations of PuLID we have found in the Internet. We appreciate the efforts of the respective developers for making PuLID accessible to a wider audience. If there are any PuLID based resources and applications that we have not mentioned here, please let us know, and we will include them in this list.

Disclaimer

This project strives to impact the domain of AI-driven image generation positively. Users are granted the freedom to create images using this tool, but they are expected to comply with local laws and utilize it responsibly. The developers do not assume any responsibility for potential misuse by users.

Citation

If PuLID is helpful, please help to ⭐ the repo.

If you find this project useful for your research, please consider citing our paper:

@article{guo2024pulid,
  title={PuLID: Pure and Lightning ID Customization via Contrastive Alignment},
  author={Guo, Zinan and Wu, Yanze and Chen, Zhuowei and Chen, Lang and He, Qian},
  journal={arXiv preprint arXiv:2404.16022},
  year={2024}
}

:e-mail: Contact

If you have any comments or questions, please open a new issue or feel free to contact Yanze Wu and Zinan Guo.

ToTheBeginning / PuLID

readme