Unilinear commited on
Commit
b81bae4
·
verified ·
1 Parent(s): 117be32

Add model card

Browse files
Files changed (1) hide show
  1. README.md +38 -0
README.md CHANGED
@@ -1,3 +1,41 @@
1
  ---
2
  license: apache-2.0
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
+ pipeline_tag: text-to-image
4
+ tags:
5
+ - diffusion
6
+ - sdxl
7
+ - pulid
8
+ - kinship-image-generation
9
  ---
10
+
11
+ # KinDiffusion
12
+
13
+ Official inference checkpoints for [KinDiffusion](https://github.com/Unilinear/KinDiffusion), a diffusion model for generating child images from two parent photos.
14
+
15
+ ## Checkpoints
16
+
17
+ | File | Purpose | Size |
18
+ | --- | --- | ---: |
19
+ | `pytorch_model_3.bin` | Identity encoder | 221 MB |
20
+ | `pytorch_model_4.bin` | KinDiffusion UNet | 12.1 GB |
21
+
22
+ Both files are required for inference. This repository contains model weights; use them with the official KinDiffusion code.
23
+
24
+ ## Download
25
+
26
+ ```bash
27
+ git clone https://github.com/Unilinear/KinDiffusion.git
28
+ cd KinDiffusion
29
+ mkdir -p output/ckpt
30
+ hf download Unilinear/KinDiffusion \
31
+ pytorch_model_3.bin pytorch_model_4.bin \
32
+ --local-dir output/ckpt
33
+ ```
34
+
35
+ Then launch the demo:
36
+
37
+ ```bash
38
+ python app_kinship.py
39
+ ```
40
+
41
+ See the [project README](https://github.com/Unilinear/KinDiffusion#readme) for environment setup and usage instructions.