赞
踩
模型的参数量和计算量参考: ThanatosShinji/onnx-tool: ONNX model's shape inference and MACs(FLOPs) counting. (github.com)这四个模型就是Stable Diffusion 1.4 最主要的4个onnx模型:
github中的百度网盘可以下载带中间tensor shape的模型. 比如:
这个模型很像BERT, 12 layers的Bert Base. 运算量6.7GMACs.
和 BertBase一样, 运算量98%都集中在MatMul上面.
这个token生成了1x77x768的hidden state需要送给UNetCondition.
这个就是UNet
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。