Instructions to use microsoft/Lens-Turbo with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use microsoft/Lens-Turbo with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("microsoft/Lens-Turbo", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
- Local Apps
- Draw Things
- DiffusionBee
File size: 24,714 Bytes
a9c8f27 b92c827 a9c8f27 b92c827 632763c 96a6303 632763c b92c827 632763c b92c827 632763c b92c827 632763c b92c827 632763c b92c827 f21b81b b92c827 d3d1930 b92c827 543c77b b92c827 96a6303 b92c827 ecb512a b92c827 96a6303 b92c827 96a6303 b92c827 96a6303 b92c827 96a6303 b92c827 aed9e78 96a6303 aed9e78 b92c827 632763c f21b81b b92c827 96a6303 b92c827 | 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 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 | ---
license: mit
language:
- en
pipeline_tag: text-to-image
---
<div align="center">
# Lens: Rethinking Training Efficiency for Foundational Text-to-Image Models
<img src="assets/teaser.webp" alt="Lens Teaser" width="100%" />
<p>
<b>Contributors (Alphabetical Order):</b><br />
<strong>Baining Guo</strong>,
<strong>Chong Luo</strong>,
<strong>Dong Chen</strong>†,
<strong>Dongdong Chen</strong>,
<strong>Fangyun Wei</strong>†,
<strong>Ji Li</strong>,
<strong>Jianmin Bao</strong>,
<strong>Jiawei Zhang</strong>*,
<strong>Jinjing Zhao</strong>*,
<strong>Lei Shi</strong>,
<strong>Qinhong Yang</strong>,
<strong>Sirui Zhang</strong>*,
<strong>Xiuyu Wu</strong>,
<strong>Xuelu Feng</strong>,
<strong>Yan Lu</strong>,
<strong>Yanchen Dong</strong>,
<strong>Yang Yue</strong>*,
<strong>Yitong Wang</strong>,
<strong>Yunuo Chen</strong>,
<strong>Zhiyang Liang</strong>*,
<strong>Ziyu Wan</strong>†
<br />
Microsoft | *Core Contributors | †Project Lead
</p>
<p>
<a href="https://arxiv.org/abs/2605.21573"><img alt="arXiv" src="https://img.shields.io/badge/arXiv-Paper-b31b1b?logo=arxiv&logoColor=white" height="22" /></a>
<a href="https://huggingface.co/microsoft/Lens"><img alt="Hugging Face" src="https://img.shields.io/badge/%F0%9F%A4%97-Models-yellow" height="22" /></a>
<a href="https://github.com/microsoft/Lens"><img alt="GitHub" src="https://img.shields.io/badge/GitHub-Repo-181717?logo=github&logoColor=white" height="22" /></a>
<a href="LICENSE"><img alt="License: MIT" src="https://img.shields.io/badge/License-MIT-green.svg" height="22" /></a>
</p>
</div>
---
**Lens** is a **3.8B-parameter** foundational text-to-image model designed for **efficient training** and **fast high-resolution generation**. It combines dense-caption pre-training, mixed-resolution learning, GPT-OSS multi-layer text features, and the FLUX.2 semantic VAE to reach competitive quality with substantially less training compute than larger T2I models.
This repository provides the minimal inference code for generating images from Lens DiT checkpoints.
## Highlights
- **Efficient Foundation** — Trained on **Lens-800M**, an 800M image-text corpus with long GPT-4.1 captions, maximizing information density per training batch.
- **Compact & Expressive** — A 48-block MMDiT denoiser leverages FLUX.2 latents and concatenated multi-layer GPT-OSS features for stronger prompt following and multilingual generalization.
- **Flexible Resolution** — Mixed-resolution training enables inference across aspect ratios from `1:2` to `2:1` and resolutions up to **1440×1440**.
- **Post-trained Variants** — RL tuning improves visual quality and artifact suppression; the distilled **Lens-Turbo** supports fast **4-step** generation.
## Gallery
<!-- LENS_GALLERY_START -->
<details name="lens-gallery" open>
<summary><b>Page 1 / 6</b> samples 000-005</summary>
<table>
<tr>
<td width="33%" valign="top">
<img src="assets/gallery/000-1440x1440.png" alt="Lens gallery sample 000" width="100%" />
<br />
<sub><b>Sample 000</b> · 1440x1440<br />A generous portion of classic British fish and chips served on a sheet of white paper, golden crispy beer-battered cod fillet alongside thick-cut chips, a wedge of lemon, mushy peas in a small dish, malt vinegar bottle nearby, wooden pub table, overhead shot</sub>
</td>
<td width="33%" valign="top">
<img src="assets/gallery/001-1440x1440.png" alt="Lens gallery sample 001" width="100%" />
<br />
<sub><b>Sample 001</b> · 1440x1440<br />The iconic Big Ben clock tower and the Houses of Parliament in London at golden hour, the River Thames reflecting warm amber light, Westminster Bridge in the foreground, a classic red double-decker bus crossing, dramatic clouds lit by sunset</sub>
</td>
<td width="33%" valign="top">
<img src="assets/gallery/002-1440x1440.png" alt="Lens gallery sample 002" width="100%" />
<br />
<sub><b>Sample 002</b> · 1440x1440<br />La Tour Eiffel au crépuscule vue depuis le Trocadéro, la structure en fer illuminée de milliers de lumières dorées scintillantes, le ciel passant du bleu profond au violet, les fontaines du Trocadéro au premier plan avec des reflets dorés, silhouettes de promeneurs</sub>
</td>
</tr>
<tr>
<td width="33%" valign="top">
<img src="assets/gallery/003-1248x1664.png" alt="Lens gallery sample 003" width="100%" />
<br />
<sub><b>Sample 003</b> · 1248x1664<br />A crystal dragon soaring through an aurora borealis sky, its entire body made of transparent faceted crystal refracting the green and purple aurora light into rainbow spectra, ice particles trailing from its wings, high fantasy digital art</sub>
</td>
<td width="33%" valign="top">
<img src="assets/gallery/004-1664x1248.png" alt="Lens gallery sample 004" width="100%" />
<br />
<sub><b>Sample 004</b> · 1664x1248<br />Aerial view of Yuanyang rice terraces in Yunnan province at sunrise, thousands of cascading water-filled paddies reflecting golden and pink sky colors, morning mist weaving between terrace layers, lush green hillside with scattered palm trees, drone photography</sub>
</td>
<td width="33%" valign="top">
<img src="assets/gallery/005-1664x1248.png" alt="Lens gallery sample 005" width="100%" />
<br />
<sub><b>Sample 005</b> · 1664x1248<br />A green iguana basking on a moss-covered fallen log in a tropical rainforest, every scale and spine rendered in sharp detail, dewdrops clinging to its skin, a blurred waterfall and lush tropical foliage in the background, National Geographic wildlife photography style</sub>
</td>
</tr>
</table>
</details>
<details name="lens-gallery">
<summary><b>Page 2 / 6</b> samples 006-011</summary>
<table>
<tr>
<td width="33%" valign="top">
<img src="assets/gallery/006-1248x1664.png" alt="Lens gallery sample 006" width="100%" />
<br />
<sub><b>Sample 006</b> · 1248x1664<br />Oil painting portrait of a Renaissance noblewoman in a deep blue velvet dress with pearl drop earrings, soft chiaroscuro lighting revealing delicate skin, craquelure texture on the painted surface, in the style of Vermeer</sub>
</td>
<td width="33%" valign="top">
<img src="assets/gallery/007-1440x1440.png" alt="Lens gallery sample 007" width="100%" />
<br />
<sub><b>Sample 007</b> · 1440x1440<br />An artisan honey jar with a hand-illustrated vintage botanical label reading "Mountain Wildflower Honey" in brown serif letterpress-style typography with decorative flourishes, detailed ink drawings of wildflowers, clover and honeybees surrounding the text, kraft paper label on clear glass jar</sub>
</td>
<td width="33%" valign="top">
<img src="assets/gallery/008-1440x1440.png" alt="Lens gallery sample 008" width="100%" />
<br />
<sub><b>Sample 008</b> · 1440x1440<br />Watercolor portrait of a thoughtful young man reading a worn leather book in a Parisian cafe, loose wet-on-wet brushstrokes bleeding into warm amber and burnt sienna washes, visible paper grain texture</sub>
</td>
</tr>
<tr>
<td width="33%" valign="top">
<img src="assets/gallery/009-1664x1248.png" alt="Lens gallery sample 009" width="100%" />
<br />
<sub><b>Sample 009</b> · 1664x1248<br />An explorer's oak desk with an aged world map spread open, a brass sextant, leather-bound navigation journal with handwritten entries, melting candle in a copper holder, scattered compass and quill pen, warm window light, still life photography</sub>
</td>
<td width="33%" valign="top">
<img src="assets/gallery/010-1664x1248.png" alt="Lens gallery sample 010" width="100%" />
<br />
<sub><b>Sample 010</b> · 1664x1248<br />New York Grand Central Terminal subway station with the classic station name "GRAND CENTRAL" spelled out in elegant white ceramic mosaic tile letters embedded in a dark green tile wall, each letter approximately eight inches tall, ornate tile border frames, the S-curve of train tracks visible</sub>
</td>
<td width="33%" valign="top">
<img src="assets/gallery/011-1664x1248.png" alt="Lens gallery sample 011" width="100%" />
<br />
<sub><b>Sample 011</b> · 1664x1248<br />A ruby-throated hummingbird hovering in front of a bright red heliconia flower, wings frozen in a figure-eight pattern showing iridescent feather detail, individual water droplets suspended around the bird, high-speed macro photography with dark background</sub>
</td>
</tr>
</table>
</details>
<details name="lens-gallery">
<summary><b>Page 3 / 6</b> samples 012-017</summary>
<table>
<tr>
<td width="33%" valign="top">
<img src="assets/gallery/012-1664x1248.png" alt="Lens gallery sample 012" width="100%" />
<br />
<sub><b>Sample 012</b> · 1664x1248<br />An old Remington typewriter with a sheet of cream-colored paper rolled into the carriage, the typed words "Chapter One: The Beginning" visible in slightly uneven Courier typeface with characteristic ink density variations, some letters slightly misaligned, warm desk lamp lighting</sub>
</td>
<td width="33%" valign="top">
<img src="assets/gallery/013-1664x1248.png" alt="Lens gallery sample 013" width="100%" />
<br />
<sub><b>Sample 013</b> · 1664x1248<br />The Great Wildebeest Migration crossing the Mara River at golden hour, hundreds of animals plunging into churning water sending spray everywhere, dust clouds rising from the riverbank, dramatic backlit scene, National Geographic documentary style</sub>
</td>
<td width="33%" valign="top">
<img src="assets/gallery/014-1248x1664.png" alt="Lens gallery sample 014" width="100%" />
<br />
<sub><b>Sample 014</b> · 1248x1664<br />A charming flower shop storefront window with hand-painted white script lettering on the glass reading "Fresh Flowers Daily" in flowing connected cursive with decorative swashes, roses and peonies arranged in buckets visible through the lettering, morning sunlight catching the painted letters</sub>
</td>
</tr>
<tr>
<td width="33%" valign="top">
<img src="assets/gallery/015-1248x1664.png" alt="Lens gallery sample 015" width="100%" />
<br />
<sub><b>Sample 015</b> · 1248x1664<br />A steampunk floating sky-city built on massive gear-driven platforms, brass and copper towers connected by chain bridges, steam-powered airships and hot air balloons docking at various levels, sunset clouds below the city, detailed concept art</sub>
</td>
<td width="33%" valign="top">
<img src="assets/gallery/016-1664x1248.png" alt="Lens gallery sample 016" width="100%" />
<br />
<sub><b>Sample 016</b> · 1664x1248<br />Milford Sound in New Zealand at dawn, a perfect mirror reflection of steep fjord walls on glass-still water, waterfalls streaming down thousand-foot cliffs, morning mist hovering above the water surface, panoramic landscape photography</sub>
</td>
<td width="33%" valign="top">
<img src="assets/gallery/017-1248x1664.png" alt="Lens gallery sample 017" width="100%" />
<br />
<sub><b>Sample 017</b> · 1248x1664<br />An Indian Bharatanatyam classical dancer in the aramandi pose, bronze ankle bells and elaborate hand mudra gestures, rich silk costume with gold temple jewelry, captured mid-performance with dramatic stage lighting</sub>
</td>
</tr>
</table>
</details>
<details name="lens-gallery">
<summary><b>Page 4 / 6</b> samples 018-023</summary>
<table>
<tr>
<td width="33%" valign="top">
<img src="assets/gallery/018-1248x1664.png" alt="Lens gallery sample 018" width="100%" />
<br />
<sub><b>Sample 018</b> · 1248x1664<br />A narrow alleyway in Marrakech's old medina with walls painted in vivid cobalt blue, colorful handwoven rugs and ceramic plates displayed along the walls, ornate wooden doors, warm sunlight from above creating dramatic shadows, Moroccan architecture</sub>
</td>
<td width="33%" valign="top">
<img src="assets/gallery/019-1664x1248.png" alt="Lens gallery sample 019" width="100%" />
<br />
<sub><b>Sample 019</b> · 1664x1248<br />A rustic wooden sign at a fishing village dock reading "Fresh Catch of the Day" in hand-carved letters painted nautical blue, thick hemp rope threaded through the sign as a border, fishing nets and lobster traps stacked in the background, seaside atmosphere</sub>
</td>
<td width="33%" valign="top">
<img src="assets/gallery/020-1664x1248.png" alt="Lens gallery sample 020" width="100%" />
<br />
<sub><b>Sample 020</b> · 1664x1248<br />A sunken shipwreck on the ocean floor completely overgrown with colorful coral formations, schools of tropical fish swimming through the broken hull and portholes, shafts of sunlight streaming down from the surface above, underwater archaeology photography</sub>
</td>
</tr>
<tr>
<td width="33%" valign="top">
<img src="assets/gallery/021-1664x1248.png" alt="Lens gallery sample 021" width="100%" />
<br />
<sub><b>Sample 021</b> · 1664x1248<br />Zhangjiajie pillar mountains rising above a sea of clouds at sunrise, golden light painting the sandstone peaks, the surreal Avatar-like floating mountain landscape stretching to the horizon, aerial drone photography capturing immense vertical scale</sub>
</td>
<td width="33%" valign="top">
<img src="assets/gallery/022-1440x1440.png" alt="Lens gallery sample 022" width="100%" />
<br />
<sub><b>Sample 022</b> · 1440x1440<br />A red-eyed tree frog perched on a bright red bromeliad flower in the Costa Rican cloud forest, its neon green body contrasting with blue-striped flanks and orange feet, water droplets on its smooth skin, extreme macro with ring flash lighting</sub>
</td>
<td width="33%" valign="top">
<img src="assets/gallery/023-1248x1664.png" alt="Lens gallery sample 023" width="100%" />
<br />
<sub><b>Sample 023</b> · 1248x1664<br />Inside a massive limestone cave, ancient stalactites and stalagmites meeting to form columns, an underground river reflecting the formations like a mirror, subtle warm lighting revealing millions of years of mineral deposits, spelunking exploration photography</sub>
</td>
</tr>
</table>
</details>
<details name="lens-gallery">
<summary><b>Page 5 / 6</b> samples 024-029</summary>
<table>
<tr>
<td width="33%" valign="top">
<img src="assets/gallery/024-1664x1248.png" alt="Lens gallery sample 024" width="100%" />
<br />
<sub><b>Sample 024</b> · 1664x1248<br />A weathered 1960s gas station with a large roadside sign reading "ROUTE 66 GAS & GO" in retro rounded sans-serif letters with a red and white color scheme, vintage gas pumps with analog dials in the foreground, a classic Chevrolet parked to the side, Americana nostalgia</sub>
</td>
<td width="33%" valign="top">
<img src="assets/gallery/025-1664x1248.png" alt="Lens gallery sample 025" width="100%" />
<br />
<sub><b>Sample 025</b> · 1664x1248<br />Construction site hoarding covered in unauthorized street art with "ART IS EVERYWHERE" spray-painted in large freehand capital letters using multiple overlapping colors of red, yellow and blue, paint drips running down from each letter, chaotic beautiful urban canvas</sub>
</td>
<td width="33%" valign="top">
<img src="assets/gallery/026-1664x1248.png" alt="Lens gallery sample 026" width="100%" />
<br />
<sub><b>Sample 026</b> · 1664x1248<br />Top-down view of a koi pond, dozens of ornamental koi fish in vivid red white orange and gold patterns swimming through crystal-clear emerald water, fallen cherry blossom petals floating on the surface, Japanese garden aerial photography</sub>
</td>
</tr>
<tr>
<td width="33%" valign="top">
<img src="assets/gallery/027-1664x1248.png" alt="Lens gallery sample 027" width="100%" />
<br />
<sub><b>Sample 027</b> · 1664x1248<br />The Potala Palace in Lhasa under a canopy of stars with the Milky Way arching overhead, Tibetan prayer wheels and butter lamps in the foreground casting warm golden light, the massive white and red palace walls glowing in moonlight, night photography</sub>
</td>
<td width="33%" valign="top">
<img src="assets/gallery/028-1248x1664.png" alt="Lens gallery sample 028" width="100%" />
<br />
<sub><b>Sample 028</b> · 1248x1664<br />Yellowstone's Grand Prismatic Spring shot from directly above by drone, concentric rings of vivid blue turquoise green yellow and orange created by thermophilic bacteria, steam rising from the surface, abstract natural color palette</sub>
</td>
<td width="33%" valign="top">
<img src="assets/gallery/029-1664x1248.png" alt="Lens gallery sample 029" width="100%" />
<br />
<sub><b>Sample 029</b> · 1664x1248<br />A herd of African elephants walking in a line across the savanna with Mount Kilimanjaro's snow-capped peak behind them, golden sunset dust kicked up by their feet creating a hazy atmosphere, telephoto wildlife photography showing massive scale</sub>
</td>
</tr>
</table>
</details>
<details name="lens-gallery">
<summary><b>Page 6 / 6</b> samples 030-031</summary>
<table>
<tr>
<td width="33%" valign="top">
<img src="assets/gallery/030-1664x1248.png" alt="Lens gallery sample 030" width="100%" />
<br />
<sub><b>Sample 030</b> · 1664x1248<br />The Hall of Mirrors at the Palace of Versailles, hundreds of candles reflected infinitely in the massive gilded mirrors, crystal chandeliers casting prismatic light across painted ceilings and gold leaf ornamentation, Baroque opulence</sub>
</td>
<td width="33%" valign="top">
<img src="assets/gallery/031-1664x1248.png" alt="Lens gallery sample 031" width="100%" />
<br />
<sub><b>Sample 031</b> · 1664x1248<br />A pirate captain's cabin, navigation charts pinned to the wall, a brass telescope and astrolabe on the desk, stacks of gold coins and a jewel-encrusted goblet, rum bottle, warm swinging lantern light casting shadows with the ship's motion</sub>
</td>
<td width="33%"></td>
</tr>
<tr>
<td width="33%"></td>
<td width="33%"></td>
<td width="33%"></td>
</tr>
</table>
</details>
<!-- LENS_GALLERY_END -->
## Installation
> **Tested environment:** Python 3.12 · CUDA 12.6 · PyTorch 2.11.0+cu126 · TorchVision 0.26.0+cu126
```bash
conda create -n lens python=3.12 -y
conda activate lens
uv pip install torch==2.11.0+cu126 torchvision==0.26.0+cu126 \
--index-url https://download.pytorch.org/whl/cu126
uv pip install -r requirements.txt
```
The default GPT-OSS encoder and FLUX.2 VAE are loaded from Hugging Face. Make sure your environment has access to any gated model repositories you use.
## Checkpoints
| Repo | Description | Steps | CFG |
| :--- | :--- | :---: | :---: |
| [`microsoft/Lens`](https://huggingface.co/microsoft/Lens) | **Default.** RL-tuned for visual quality | 20 | 5.0 |
| [`microsoft/Lens-Turbo`](https://huggingface.co/microsoft/Lens-Turbo) | Distilled from the RL model for fast 4-step sampling | 4 | 1.0 |
| [`microsoft/Lens-Base`](https://huggingface.co/microsoft/Lens-Base) | Supervised base model (no RL, no distillation) | 50 | 5.0 |
Pick a variant by passing its repo id to `--repo_id` (CLI) or `LensPipeline.from_pretrained(...)` (Python).
## Inference
> **Important:** run from the cloned repo root so `from lens import LensPipeline` resolves to this package — importing `lens` is what registers `LensGptOssEncoder` / `LensTransformer2DModel` with the `transformers` and `diffusers` namespaces that `model_index.json` references.
**Python API:**
```python
import torch
from lens import LensPipeline
pipe = LensPipeline.from_pretrained(
"microsoft/Lens", torch_dtype=torch.bfloat16
).to("cuda")
image = pipe(
prompt="A cat holding a sign that says \"hello world\"",
base_resolution=1440, aspect_ratio="1:1",
num_inference_steps=20, guidance_scale=5.0,
generator=torch.Generator("cuda").manual_seed(0),
).images[0]
image.save("lens.png")
```
To trade speed for VRAM, replace `.to("cuda")` with `pipe.enable_model_cpu_offload()`.
**CLI — basic usage:**
```bash
python inference.py \
--repo_id "microsoft/Lens" \
--prompt "A cinematic mountain lake at sunrise, soft mist, detailed reflections" \
--base_resolution 1440 --aspect_ratio 1:1 \
--steps 20 --cfg 5.0 --n 1 --seed 42 \
--out ./outputs
```
**Batch generation** — join multiple prompts with `|`:
```bash
python inference.py \
--repo_id "microsoft/Lens" \
--steps 20 --cfg 5.0 \
--prompt "a red fox in snow|a glass greenhouse at night"
```
**A100 / V100 (no MXFP4 kernels)** — dequantize the GPT-OSS encoder to bf16:
```bash
python inference.py \
--repo_id "microsoft/Lens" \
--steps 20 --cfg 5.0 \
--prompt "a cat" \
--disable_mxfp4 --offload
```
### Options
| Flag | Description | Default |
| :--- | :--- | :--- |
| `--repo_id` | HF repo id (or local path) of the assembled Lens pipeline | `microsoft/Lens` |
| `--base_resolution` | `1024` or `1440` | `1440` |
| `--aspect_ratio` | `1:2`, `9:16`, `2:3`, `3:4`, `1:1`, `4:3`, `3:2`, `16:9`, `2:1` | `1:1` |
| `--steps` | Number of denoising steps | `20` |
| `--cfg` | Classifier-free guidance scale | `5.0` |
| `--n` | Number of images per prompt | `1` |
| `--seed` | Random seed (omit for non-deterministic) | — |
| `--out` | Output directory | `./outputs` |
| `--dtype` | Compute dtype: `bfloat16`, `float16`, `float32` | `bfloat16` |
| `--disable_mxfp4` | Dequantize the GPT-OSS text encoder to `--dtype` (required on A100 / V100; Hopper+ keeps MXFP4 by default for less VRAM) | — |
| `--offload` | Enable diffusers CPU offload (`text_encoder->transformer->vae`) to reduce peak VRAM | — |
| `--reasoner` | Refine prompts with the loaded GPT-OSS encoder before generation | — |
| `--api_url` / `--api_key` / `--api_model` | Use an OpenAI-compatible API for prompt refinement (takes precedence over `--reasoner`) | — |
## Citation
```bibtex
@article{zhao2026lens,
title = {Lens: Rethinking Training Efficiency for Foundational Text-to-Image Models},
author = {Guo, Baining and Luo, Chong and Chen, Dong and Chen, Dongdong and Wei, Fangyun and Li, Ji and Bao, Jianmin and Zhang, Jiawei and Zhao, Jinjing and Shi, Lei and Yang, Qinhong and Zhang, Sirui and Wu, Xiuyu and Feng, Xuelu and Lu, Yan and Dong, Yanchen and Yue, Yang and Wang, Yitong and Chen, Yunuo and Liang, Zhiyang and Wan, Ziyu},
journal = {arXiv preprint arXiv:2605.21573},
year = {2026}
}
```
## Responsible AI
The model is released for research purposes only and is not intended for product or service deployment. Responsible AI considerations were incorporated throughout the development process, including data selection, model training, and evaluation.
The training data includes a combination of public, licensed, and internal datasets that were processed to remove clearly identifiable personal information and reduce harmful content where possible. However, as the data is largely sourced from web-scale collections, it may contain biases or uneven representation. As a result, the model may generate outputs that are inaccurate, biased, or inappropriate under certain prompts, including content that could be misleading or raise copyright or IP-related concerns.
Given these limitations, the model should be used in controlled research settings, with appropriate human oversight. Downstream users are responsible for applying additional safeguards, such as content moderation, validation, and compliance checks, before using the model in broader applications.
## Privacy
This project does not collect any usage data. For more information, see the [Microsoft Privacy Statement](https://go.microsoft.com/fwlink/?LinkId=521839).
## License
This project is released under the [MIT License](LICENSE).
|