ONNX Model Zoo

amd/segmind-vega-amdnpu

By amd · ↓ 0 · ♥ 2 · Updated 2026-02-25

What's in this repo

This page lists the files in amd/segmind-vega-amdnpu, including text_encoder_fp16.onnx, text_encoder_2_fp16.onnx, replaced.onnx and more.

  • 5 ONNX models (.onnx files)
  • 44 files, including 15 .json, 5 .onnx, 5 .txt, 3 .safetensors, 3 .ctrlpkt, 3 .fconst
  • Last updated 2026-02-25

ONNX files

File Size Download
text_encoder/text_encoder_fp16.onnx — Download
text_encoder_2/text_encoder_2_fp16.onnx — Download
unet/dd/replaced.onnx — Download
vae_decoder/dd/replaced.onnx — Download
vae_encoder/dd/replaced.onnx — Download

How to use

These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.

README


license: apache-2.0 language: en tags:

  • text-to-image
  • ultra-realistic
  • stable-diffusion
  • distilled-model
  • knowledge-distillation pinned: true datasets:
  • zzliang/GRIT
  • wanng/midjourney-v5-202304-clean

🚀 Segmind-Vega on AMD NPU

"Segmind Vega is a distilled version of the Stable Diffusion XL (SDXL), offering a 70% reduction in size and an impressive 100% speedup while retaining high-quality text-to-image generation capabilities. This makes it an ideal choice for applications that require efficient and fast image generation." (segmind/Segmind-Vega)

This version of the model has been optimized to run on AMD NPUs.

For more information, refer to the original model card under segmind/Segmind-Vega.

Model Details

Model Details Description
Person or organization developing model AMD
Original model authors Segmind
Model date February 2026
Model version 1.7.1
Model type Diffusion-based text-to-image generation model
Information about training algorithms, parameters, fairness constraints or other applied approaches, and features This is a model that can be used to generate and modify images based on text prompts.
License Apache 2.0
Where to send questions or comments about the model Community Tab and AMD Developer Community Discord

⚡ Intended Use

Getting Started

To get started with this model, visit github.com/amd/sd-sandbox.

⚓ Ethical Considerations

AMD is committed to conducting our business in a fair, ethical and honest manner and in compliance with all applicable laws, rules and regulations. You can find out more at the AMD Ethics and Compliance page.

⚠️ Caveats and Recommendations

Please visit the original model card for more details: segmind/Segmind-Vega.

Model info

onnx safetensors text-to-image ultra-realistic stable-diffusion distilled-model knowledge-distillation en dataset:zzliang/GRIT dataset:wanng/midjourney-v5-202304-clean license:apache-2.0 region:us
↓ 0 ♥ 2

Created: 2026-02-05

Updated: 2026-02-25

View on Hugging Face

Files (44)

.gitattributes
LICENSE
README.md
scheduler/scheduler_config.json
text_encoder/config.json
text_encoder/model.safetensors
text_encoder/text_encoder_fp16.onnx ONNX
text_encoder_2/config.json
text_encoder_2/model.fp16.safetensors
text_encoder_2/model.safetensors
text_encoder_2/text_encoder_2_fp16.onnx ONNX
text_encoder_2/text_encoder_2_fp16.onnx.data
tokenizer/merges.txt
tokenizer/special_tokens_map.json
tokenizer/tokenizer_config.json
tokenizer/vocab.json
tokenizer_2/merges.txt
tokenizer_2/special_tokens_map.json
tokenizer_2/tokenizer_config.json
tokenizer_2/vocab.json
unet/config.json
unet/dd/cache/NhwcConv_0-unetconv_inConv_meta.json
unet/dd/dd_metastate_SD15_Unet_NhwcConv_0-unetconv_inConv.ctrlpkt
unet/dd/dd_metastate_SD15_Unet_NhwcConv_0-unetconv_inConv.fconst
unet/dd/dd_metastate_SD15_Unet_NhwcConv_0-unetconv_inConv.state
unet/dd/dd_metastate_SD15_Unet_NhwcConv_0-unetconv_inConv.super
unet/dd/onnx_report.txt
unet/dd/replaced.onnx ONNX
vae_decoder/config.json
vae_decoder/dd/cache/NhwcConv_0-post_quant_convConv_meta.json
vae_decoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-post_quant_convConv.ctrlpkt
vae_decoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-post_quant_convConv.fconst
vae_decoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-post_quant_convConv.state
vae_decoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-post_quant_convConv.super
vae_decoder/dd/onnx_report.txt
vae_decoder/dd/replaced.onnx ONNX
vae_encoder/config.json
vae_encoder/dd/cache/NhwcConv_0-encoderconv_inConv_meta.json
vae_encoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-encoderconv_inConv.ctrlpkt
vae_encoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-encoderconv_inConv.fconst
vae_encoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-encoderconv_inConv.state
vae_encoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-encoderconv_inConv.super
vae_encoder/dd/onnx_report.txt
vae_encoder/dd/replaced.onnx ONNX

Related models