File size: 3,745 Bytes
670d192 b563fbd 007f691 670d192 b563fbd 670d192 b563fbd 670d192 b563fbd 670d192 |
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 |
---
license: apache-2.0
library_name: diffusers
pipeline_tag: text-to-image
base_model: shallowdream204/BitDance-14B-64x
language:
- en
tags:
- bitdance
- text-to-image
- custom-pipeline
- diffusers
- qwen
---
# BitDance-14B-64x (Diffusers)
Diffusers-converted checkpoint for BitDance-14B-64x with bundled custom pipeline code (`bitdance_diffusers`) for direct loading with `DiffusionPipeline`.
## Quickstart (native diffusers)
```python
import torch
from diffusers import DiffusionPipeline
# Local path (recommended - no trust_remote_code needed)
model_path = "BiliSakura/BitDance-14B-64x-diffusers"
pipe = DiffusionPipeline.from_pretrained(
model_path,
custom_pipeline=model_path,
torch_dtype=torch.bfloat16,
).to("cuda")
result = pipe(
prompt = "A close-up portrait in a cinematic photography style, capturing a girl-next-door look on a sunny daytime urban street. She wears a khaki sweater, with long, flowing hair gently draped over her shoulders. Her head is turned slightly, revealing soft facial features illuminated by realistic, delicate sunlight coming from the left. The sunlight subtly highlights individual strands of her hair. The image has a Canon film-like color tone, evoking a warm nostalgic atmosphere.",
height=1024,
width=1024,
num_inference_steps=50,
guidance_scale=7.5,
)
result.images[0].save("bitdance_14b_64x.png")
```
## Test Running
Run tests from the model directory in your active Python environment:
```bash
python test_bitdance.py
```
## VRAM Usage by Resolution
Measured on `NVIDIA A100-SXM4-80GB` using:
- `dtype=torch.bfloat16`
- `num_inference_steps=30`
- `guidance_scale=7.5`
- prompt: `A close-up portrait in a cinematic photography style, capturing a girl-next-door look on a sunny daytime urban street. She wears a khaki sweater, with long, flowing hair gently draped over her shoulders. Her head is turned slightly, revealing soft facial features illuminated by realistic, delicate sunlight coming from the left. The sunlight subtly highlights individual strands of her hair. The image has a Canon film-like color tone, evoking a warm nostalgic atmosphere.`
| Resolution | Peak Allocated VRAM (GiB) | Peak Reserved VRAM (GiB) | Time (s) | Status |
| --- | ---: | ---: | ---: | --- |
| 512x512 | 39.60 | 40.62 | 4.08 | ok |
| 1024x1024 | 41.21 | 50.15 | 15.79 | ok |
| 1280x768 | 40.88 | 49.52 | 14.78 | ok |
| 768x1280 | 40.88 | 49.52 | 14.75 | ok |
| 1536x640 | 40.88 | 49.52 | 14.76 | ok |
| 2048x512 | 41.21 | 50.15 | 15.85 | ok |
## Model Metadata
- Pipeline class: `BitDanceDiffusionPipeline`
- Diffusers version in config: `0.36.0`
- Parallel prediction factor: `64`
- Text stack: `Qwen3ForCausalLM` + `Qwen2TokenizerFast`
- Supported resolutions include `1024x1024`, `1280x768`, `768x1280`, `2048x512`, and more (see `model_index.json`)
## Citation
If you use this model, please cite BitDance and Diffusers:
```bibtex
@article{ai2026bitdance,
title = {BitDance: Scaling Autoregressive Generative Models with Binary Tokens},
author = {Ai, Yuang and Han, Jiaming and Zhuang, Shaobin and Hu, Xuefeng and Yang, Ziyan and Yang, Zhenheng and Huang, Huaibo and Yue, Xiangyu and Chen, Hao},
journal = {arXiv preprint arXiv:2602.14041},
year = {2026}
}
@inproceedings{von-platen-etal-2022-diffusers,
title = {Diffusers: State-of-the-art diffusion models},
author = {Patrick von Platen and Suraj Patil and Anton Lozhkov and Damar Jablonski and Hernan Bischof and Thomas Wolf},
booktitle = {GitHub repository},
year = {2022},
url = {https://github.com/huggingface/diffusers}
}
```
## License
This repository is distributed under the Apache-2.0 license, consistent with the upstream BitDance release.
|