diff --git a/api/extras.json b/api/extras.json index fb772bac..2c1dc374 100644 --- a/api/extras.json +++ b/api/extras.json @@ -1,7 +1,9 @@ { "diffusion": [ ["diffusion-knollingcase", "Aybeeceedee/knollingcase"], - ["diffusion-openjourney", "prompthero/openjourney"] + ["diffusion-openjourney", "prompthero/openjourney"], + ["diffusion-stably-diffused-onnx-v2-6", "../models/tensors/stablydiffuseds_26.safetensors"], + ["diffusion-unstable-ink-dream-onnx-v6", "../models/tensors/unstableinkdream_v6.safetensors"] ], "correction": [], "upscaling": [] diff --git a/api/onnx_web/convert/__main__.py b/api/onnx_web/convert/__main__.py index fe588036..13e6d7a6 100644 --- a/api/onnx_web/convert/__main__.py +++ b/api/onnx_web/convert/__main__.py @@ -43,9 +43,6 @@ base_models: Models = { ), # TODO: should have its own converter ("upscaling-stable-diffusion-x4", "stabilityai/stable-diffusion-x4-upscaler"), - # TODO: testing safetensors - ("diffusion-stably-diffused-onnx-v2-6", "../models/tensors/stablydiffuseds_26.safetensors"), - ("diffusion-unstable-ink-dream-onnx-v6", "../models/tensors/unstableinkdream_v6.safetensors"), ], "correction": [ (