File size: 4,311 Bytes
0b3246a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
---
license: apache-2.0
---


# πŸš€ ComfyUI_InfiniteYou
An implementation of **InfiniteYou** for **ComfyUI**

Native support for [InfiniteYou](https://github.com/bytedance/InfiniteYou) in [ComfyUI](https://github.com/comfyanonymous/ComfyUI), designed by the ZenAI team.  


**✨ Support further development by starring the project! ✨**


![teaser](./teaser.jpg)

---

## πŸ”₯ News
- **[03/2025]** πŸ”₯ Integrate Face Combine feature to predict future children
- **[03/2025]** πŸ”₯ Code updated and released as the first version.

---

## πŸ“œ Introduction  

πŸš€ **InfiniteYou** is a **SOTA zero-shot identity preservation** model by **ByteDance**, built on **FLUX**. This repo brings it to **ComfyUI**, powered by **ZenAI**.  

πŸ”— **References:**  
πŸ“„ [Paper](https://arxiv.org/abs/2503.16418) | πŸ’Ύ [Official Repo](https://github.com/bytedance/InfiniteYou)  

πŸ’‘ Inspired by **ZenID** πŸ”— [ZenID Repo](https://github.com/vuongminh1907/ComfyUI_ZenID)  

πŸ”₯ Stay tuned for updates!  

---

## πŸ† Model Zoo

The main author has released two versions of the model, each tailored for a specific purpose:  

- πŸ”Ή **sim_stage1** – Prioritizes **higher identity similarity** for more accurate face preservation.  
- 🎨 **aes_stage2** – Focuses on **better text-image alignment** and enhanced **aesthetics**.  

To ensure seamless integration with **ComfyUI**, we have converted the model to the **safetensors** format. 


**Download the model on Hugging Face:**  
πŸ‘‰ [ComfyUI_InfiniteYou](https://huggingface.co/vuongminhkhoi4/ComfyUI_InfiniteYou) 
---

## πŸ› οΈ Workflow
### **Zero-Shot Task**
![Musk](./musk.png)

### **FaceCombine Task**
![Children](./face_combine_workflow.png)
---

## πŸ“¦ Installation

### Step 1: Clone the Repo
```bash
cd custom_nodes
git clone https://github.com/ZenAI-Comfy/ComfyUI_InfiniteYou
```
### Step 2: Install Requirements
```bash
cd ComfyUI_InfiniteYou
pip install -r requirements.txt
```

### Step 3: Download ControlNet Models
```bash
cd ../../models/controlnet
wget https://huggingface.co/vuongminhkhoi4/ComfyUI_InfiniteYou/resolve/main/aes_stage2_control_net/aes_stage2_control.safetensors
wget https://huggingface.co/vuongminhkhoi4/ComfyUI_InfiniteYou/resolve/main/sim_stage1_control_net/sim_stage1_control_net.safetensors
cd ..
```
### Step 4: Download Image Projection Files
```bash
mkdir InfiniteYou
cd InfiniteYou
wget https://huggingface.co/vuongminhkhoi4/ComfyUI_InfiniteYou/resolve/main/aes_stage2_control_net/aes_stage2_img_proj.bin
wget https://huggingface.co/vuongminhkhoi4/ComfyUI_InfiniteYou/resolve/main/sim_stage1_control_net/sim_stage1_img_proj.bin
cd ../..
```
### Step 5: Download InsightFace model
The InsightFace model is **antelopev2** (not the classic buffalo_l). Download the models (for example from [here](https://drive.google.com/file/d/18wEUfMNohBJ4K3Ly5wpTejPfDzp-8fI8/view?usp=sharing) or [here](https://huggingface.co/MonsterMMORPG/tools/tree/main)), unzip and place them in the `ComfyUI/models/insightface/models/antelopev2` directory.

Alternatively, you can quickly download all **models** using the following command:
```
# make sure you are in the ComfyUI directory
cd custom_nodes/
git clone https://github.com/ZenAI-Vietnam/ComfyUI_InfiniteYou
python ComfyUI_InfiniteYou/downloadmodel.py
pip install -r ComfyUI_InfiniteYou/requirements.txt
```

## 🧭 Usage

πŸ”Ή For `aes_stage2`: Try file `aes_stages2.json` in `workflows`

πŸ”Ή For `sim_stage1`: Try file `sim_stages1.json` in `workflows`

Predict your future children
πŸ”Ή For `Face Combine`: Try file `face_combine.json` in `workflows`



## πŸ“ž Contact for Work 🌟
This implementation of InfiniteYou is brought to you by the ZenAI Team.

<img src="https://huggingface.co/vuongminhkhoi4/ComfyUI_InfiniteYou/blob/main/zenai.png" width="400" />

If you need more polished and enhanced version, please contact us through:  
- πŸ“± **Facebook Page**: [ZenAI](https://web.facebook.com/zenai.vn)  
- ☎️ **Phone**: 0971912713 Miss. Chi 

## πŸ“– Citation

```bibtex
@article{jiang2025infiniteyou,
  title={{InfiniteYou}: Flexible Photo Recrafting While Preserving Your Identity},
  author={Jiang, Liming and Yan, Qing and Jia, Yumin and Liu, Zichuan and Kang, Hao and Lu, Xin},
  journal={arXiv preprint},
  volume={arXiv:2503.16418},
  year={2025}
}
```