nielsr HF Staff commited on
Commit
cf08f0f
·
verified ·
1 Parent(s): 0c120b4

Update model card metadata, links and usage instructions

Browse files

Hi, I'm Niels from the community science team at Hugging Face. I've updated your model card to improve its visibility and usability:
- Updated placeholder badges for arXiv and ICML 2026 with the actual project links.
- Added `base_model` and `datasets` metadata to link this repository to the FLUX model and the SA-Z dataset.
- Included the CLI and Streamlit inference instructions from your GitHub README to provide a "Quick Start" guide for users.
- Linked the paper to the Hugging Face papers page.

Files changed (1) hide show
  1. README.md +16 -21
README.md CHANGED
@@ -1,14 +1,18 @@
1
  ---
2
- license: mit
 
3
  library_name: diffusers
 
4
  pipeline_tag: text-to-image
 
 
 
5
  tags:
6
  - occlusion
7
  - layout
8
  - flux
9
- language:
10
- - en
11
  ---
 
12
  <table align="center" cellspacing="0" cellpadding="0" style="margin:0 auto;">
13
  <tr>
14
  <td valign="middle" style="padding-right:10px;">
@@ -23,9 +27,9 @@ language:
23
 
24
  <div align="center" style="margin-top:14px;">
25
  <a href='https://henghuiding.com/OcclusionFormer/'><img src='https://img.shields.io/badge/Project-Page-Green'></a> &nbsp;&nbsp;&nbsp;&nbsp;
26
- <a href='https://icml.cc/Downloads/2026'><img src='https://img.shields.io/badge/ICML-2026-blue'></a> &nbsp;&nbsp;&nbsp;&nbsp;
27
- <a href='https://arxiv.org/'><img src='https://img.shields.io/badge/arXiv-Coming%20Soon-b31b1b'></a> &nbsp;&nbsp;&nbsp;&nbsp;
28
- <a href='https://huggingface.co/'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20Weights-Hugging Face%20-orange'></a> &nbsp;&nbsp;&nbsp;&nbsp;
29
  <a href='https://huggingface.co/datasets/FudanCVL/SA-Z'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20SA--Z-Hugging Face%20-orange'></a>
30
  </div>
31
  <p align="center" style="margin:4px 0 0 0;">
@@ -36,7 +40,12 @@ language:
36
  <p align="center" style="margin:1px 0 0 0; font-size:1.48em; font-weight:900; color:#ff6a00;">ICML 2026</p>
37
  <p align="center" style="margin:1px 0 0 0; font-size:1.08em; color:#6b7280;"><em>✉ Corresponding Author</em></p>
38
 
 
 
 
 
39
  ## 🔥 News
 
40
  - [2026/05/18] Release **inference code**, **model weights** and **SA-Z dataset**.
41
  - [2026/05/18] Release **OcclusionFormer open-source package** in this repository.
42
  - [2026/4/30] OcclusionFormer is accepted to **ICML 2026**.
@@ -85,9 +94,7 @@ pip install --upgrade -r requirements.txt
85
 
86
  3. Download checkpoint
87
 
88
- ```bash
89
- https://huggingface.co/FudanCVL/OcclusionFormer
90
- ```
91
 
92
  4. Run Streamlit demo
93
 
@@ -107,18 +114,6 @@ python inference_occlusionformer.py \
107
  --overwrite
108
  ```
109
 
110
- Batch inference with a directory of JSON layouts:
111
-
112
- ```bash
113
- python inference_occlusionformer.py \
114
- --model_path /path/to/FLUX.1-dev \
115
- --ckpt_path /path/to/occlusionformer_checkpoint_dir \
116
- --layout_dir ./examples \
117
- --output_dir ./outputs_occlusionformer \
118
- --enable_layout \
119
- --overwrite
120
- ```
121
-
122
  ---
123
  ## ✅ TODO
124
 
 
1
  ---
2
+ language:
3
+ - en
4
  library_name: diffusers
5
+ license: mit
6
  pipeline_tag: text-to-image
7
+ base_model: black-forest-labs/FLUX.1-dev
8
+ datasets:
9
+ - FudanCVL/SA-Z
10
  tags:
11
  - occlusion
12
  - layout
13
  - flux
 
 
14
  ---
15
+
16
  <table align="center" cellspacing="0" cellpadding="0" style="margin:0 auto;">
17
  <tr>
18
  <td valign="middle" style="padding-right:10px;">
 
27
 
28
  <div align="center" style="margin-top:14px;">
29
  <a href='https://henghuiding.com/OcclusionFormer/'><img src='https://img.shields.io/badge/Project-Page-Green'></a> &nbsp;&nbsp;&nbsp;&nbsp;
30
+ <a href='https://icml.cc/virtual/2026/poster/66159'><img src='https://img.shields.io/badge/ICML-2026-blue'></a> &nbsp;&nbsp;&nbsp;&nbsp;
31
+ <a href='https://arxiv.org/abs/2605.21343'><img src='https://img.shields.io/badge/arXiv-2605.21343-b31b1b.svg'></a> &nbsp;&nbsp;&nbsp;&nbsp;
32
+ <a href='https://huggingface.co/papers/2605.21343'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20Paper-Hugging Face-orange'></a> &nbsp;&nbsp;&nbsp;&nbsp;
33
  <a href='https://huggingface.co/datasets/FudanCVL/SA-Z'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20SA--Z-Hugging Face%20-orange'></a>
34
  </div>
35
  <p align="center" style="margin:4px 0 0 0;">
 
40
  <p align="center" style="margin:1px 0 0 0; font-size:1.48em; font-weight:900; color:#ff6a00;">ICML 2026</p>
41
  <p align="center" style="margin:1px 0 0 0; font-size:1.08em; color:#6b7280;"><em>✉ Corresponding Author</em></p>
42
 
43
+ This repository contains the weights for **OcclusionFormer**, as presented in the paper [OcclusionFormer: Arranging Z-Order for Layout-Grounded Image Generation](https://huggingface.co/papers/2605.21343).
44
+
45
+ The official code is available at: [https://github.com/FudanCVL/OcclusionFormer](https://github.com/FudanCVL/OcclusionFormer)
46
+
47
  ## 🔥 News
48
+ - [2026/05/21] Release **OcclusionFormer** on [ArXiv](https://arxiv.org/abs/2605.21343)!
49
  - [2026/05/18] Release **inference code**, **model weights** and **SA-Z dataset**.
50
  - [2026/05/18] Release **OcclusionFormer open-source package** in this repository.
51
  - [2026/4/30] OcclusionFormer is accepted to **ICML 2026**.
 
94
 
95
  3. Download checkpoint
96
 
97
+ Weights can be found in this repository. Download `occlusionformer` to `./ckpt`.
 
 
98
 
99
  4. Run Streamlit demo
100
 
 
114
  --overwrite
115
  ```
116
 
 
 
 
 
 
 
 
 
 
 
 
 
117
  ---
118
  ## ✅ TODO
119