apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

move tensors to extras file

This commit is contained in:
Sean Sube 2023-02-08 22:38:19 -06:00
parent 4f71348f98
commit 511143eea5
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/extras.json
View File

@ -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": []

3
api/onnx_web/convert/__main__.py
View File

@ -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": [
(