chaseSun123/ernie-image-ncnn

ERNIE-Image ncnn 模型

这是从 Baidu/ERNIE-Image-Turbo 转换的 ncnn 格式模型,用于纯本地文字生图(不依赖 PyTorch)。

使用

参考 GitHub 仓库:touqiang-guang/ernie-image-ncnn

文件说明

• te_seg0..4.ncnn.bin/.param:文本编码器(5 段)
• dit_segA / dit_segB0..8 / dit_segZ:扩散模型 DiT(11 段)
• ernie_vae_decoder.ncnn.bin/.param:VAE 解码器
• time_embed_weight.f32 / text_proj_weight.f32:时间嵌入/文本投影辅助权重
• tokenizer/:BPE 分词器

转换方法

使用 pnnx 从 ERNIE-Image-Turbo safetensors 转换,详见 GitHub 仓库文档。

来源

原始模型:Baidu/ERNIE-Image-Turbo