返回模型
说明文档
适用于 ONNX Runtime CUDA 提供程序的 Stable Diffusion 1.5
简介
本仓库托管了 Stable Diffusion 1.5 的优化 ONNX 模型,旨在利用 ONNX Runtime CUDA 执行提供程序在 Nvidia GPU 上加速推理。该模型无法在 CPU 或 DirectML 等其他执行提供程序上运行。
这些模型由 Olive 通过如下命令生成:
python stable_diffusion.py --provider cuda --optimize
模型描述
- 开发者: Stability AI
- 模型类型: 基于扩散的文本生成图像模型
- 许可证: The CreativeML OpenRAIL M license
- 模型描述: 这是由 runwayml/stable-diffusion-v1-5 模型转换而来,用于在 ONNX Runtime 上配合 CUDA 执行提供程序进行推理。
tlwu/stable-diffusion-v1-5-onnxruntime
作者 tlwu
text-to-image
↓ 0
♥ 4
创建时间: 2023-12-30 08:29:40+00:00
更新时间: 2023-12-31 06:33:50+00:00
在 Hugging Face 上查看文件 (14)
.gitattributes
README.md
feature_extractor/preprocessor_config.json
model_index.json
safety_checker/model.onnx
ONNX
scheduler/scheduler_config.json
text_encoder/model.onnx
ONNX
tokenizer/merges.txt
tokenizer/special_tokens_map.json
tokenizer/tokenizer_config.json
tokenizer/vocab.json
unet/model.onnx
ONNX
vae_decoder/model.onnx
ONNX
vae_encoder/model.onnx
ONNX