search

microsoft / onnxruntime-training-examples

Examples for using ONNX Runtime for model training.
MIT License
310 stars 62 forks source link

add stable diffusion inference #149

Closed prathikr closed 1 year ago

prathikr commented 1 year ago

adds code to load a UNet trained via ONNX Runtime into a Stable Diffusion Pipeline

prathikr commented 1 year ago

Successful training runs: https://ml.azure.com/experiments/id/18bb2273-8552-44fb-915f-9051f19b0cb9?wsid=/subscriptions/ed2cab61-14cc-4fb3-ac23-d72609214cfd/resourcegroups/AMLDataCache/workspaces/datacachetest&tid=72f988bf-86f1-41af-91ab-2d7cd011db47