趋近智
PixArt-Σ XL 2 1024-MS
参数
610M
模态
Image
辅助参数
4.78B
架构
Diffusion Transformer
许可证
Apache 2.0
发布日期
10 Apr 2024
API 价格 (每 1M)
仅支持自建托管
硬件选型
精度
工作负载显存: 2.91 GB
INT4 · 1024x1024
DiT 骨干权重: 0.28 GB
VAE 解码工作区: 0.12 GB
运行时与计算图: 2.50 GB
最小所需硬件
1x NVIDIA L4 24GB
2.9 / 21.6 GB usable
1x RTX 3060 12GB
2.9 / 10.8 GB usable
Apple M4 24GB
2.9 / 18.0 GB usable
确定特定工作负载的硬件需求：
在显存计算器中打开
架构图
关于 PixArt-Σ XL 2 1024-MS
PixArt-Σ is an enhanced 610M diffusion transformer introducing weak-to-strong training and KV-compression DiT blocks to support native 4K image generation at low compute cost.
技术规格
注意力
注意力头
16
注意力头维度
72
维度
辅助参数
4.78B
隐藏维度大小
1,152
层数
28