ApX 标志ApX 标志

趋近智

AuraFlow v0.3

参数

6.8B

模态

Image

辅助参数

4.78B

架构

Diffusion Transformer

许可证

Apache 2.0

发布日期

25 Jul 2024

API 价格 (每 1M)

仅支持自建托管

硬件选型

精度

工作负载显存: 6.10 GB

INT4 · 1024x1024

DiT 骨干权重: 3.17 GB

VAE 解码工作区: 0.44 GB

运行时与计算图: 2.50 GB

最小所需硬件

1x NVIDIA L4 24GB

6.1 / 21.6 GB usable

1x RTX 3060 12GB

6.1 / 10.8 GB usable

Apple M4 24GB

6.1 / 18.0 GB usable

确定特定工作负载的硬件需求:

在显存计算器中打开

架构图

Noisy Image LatentsLatent Image Space (VAE)Patch Embedding2D Latent PatchifyHidden: 2k · Text: 4.78B enc · Timestep (t)x 28 DiT blocksAdaLN ModulationNorm · Timestep scale/shiftJoint / Cross-Attention12 headsHead dim: 256+AdaLN ModulationNorm · Timestep scale/shiftFeed-Forward (MLP)Activation+Final Norm (AdaLN Scale)Predicted Noise / LatentsDenoised Latents to VAE

关于 AuraFlow v0.3

AuraFlow v0.3 is an open-source 6.8B parameter rectified-flow diffusion transformer trained by fal.ai, featuring dual-stream attention layers with UMT5-XXL conditioning.

技术规格

注意力

注意力头

12

注意力头维度

256

维度

辅助参数

4.78B

隐藏维度大小

2,048

层数

28

关于 AuraFlow


其他 AuraFlow 模型