ONNX Runtime是一个支持多种语言和平台的生产级人工智能引擎。它利用现有的技术堆栈优化训练和推理,并实现生成人工智能和大模型训练。
import os os.environ["HF_ENDPOINT"] = "https://hf-mirror.com" from optimum.onnxruntime import ORTStableDiffusionXLPipeline model_id = "..\\..\\stabilityai\\stable-diffusion-xl-base-1.0" # model_id = "stabilityai\\stable-diffusion-xl-base-1.0" pipeline = ORTStableDiffusionXLPipeline.from_pretrained(model_id) prompt = "sailing ship in storm by Leonardo da Vinci" image = pipeline(prompt).images[0] image.show()