ApX 标志ApX 标志

趋近智

Stable Diffusion 2.1

参数

860M

模态

Image

辅助参数

200M

架构

Diffusion Transformer

许可证

CreativeML OpenRAIL-M

发布日期

7 Dec 2022

API 价格 (每 1M)

仅支持自建托管

硬件选型

精度

工作负载显存: 2.93 GB

INT4 · 1024x1024

DiT 骨干权重: 0.40 GB

VAE 解码工作区: 0.03 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

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

在显存计算器中打开

架构图

Noisy Image LatentsLatent Image Space (VAE)Patch Embedding2D Latent PatchifyHidden: 1k · Text: 200M enc · Timestep (t)x 8 DiT blocksAdaLN ModulationNorm · Timestep scale/shiftJoint / Cross-Attention51 headsHead dim: 20+AdaLN ModulationNorm · Timestep scale/shiftFeed-Forward (MLP)Activation+Final Norm (AdaLN Scale)Predicted Noise / LatentsDenoised Latents to VAE

关于 Stable Diffusion 2.1

Stable Diffusion 2.1 is an 865M parameter latent diffusion UNet model using OpenCLIP ViT-H/14 text encoder, fine-tuned on non-standard aesthetic subsets for native 768x768 resolution.

技术规格

注意力

注意力头

51

注意力头维度

20

维度

辅助参数

200M

隐藏维度大小

1,024

层数

8

关于 Stable Diffusion 2


其他 Stable Diffusion 2 模型