apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): change weights filename for new models to match optimum

This commit is contained in:
Sean Sube 2023-12-23 22:18:35 -06:00
parent 14208de393
commit ebe813d035
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/constants.py
View File

@ -1,5 +1,5 @@
ONNX_MODEL = "model.onnx"
ONNX_WEIGHTS = "weights.pb"
ONNX_WEIGHTS = "model.onnx_data"
LATENT_FACTOR = 8
LATENT_CHANNELS = 4

2
api/onnx_web/convert/diffusion/diffusion.py
View File

@ -865,7 +865,7 @@ def convert_diffusion_diffusers_optimum(
unet_path,
save_as_external_data=True,
all_tensors_to_one_file=True,
location="weights.pb",
location=ONNX_WEIGHTS,
)
return (True, dest_path)

4
api/onnx_web/convert/diffusion/diffusion_xl.py
View File

@ -9,7 +9,7 @@ from onnx.shape_inference import infer_shapes_path
from onnxruntime.transformers.float16 import convert_float_to_float16
from optimum.exporters.onnx import main_export
from ...constants import ONNX_MODEL
from ...constants import ONNX_MODEL, ONNX_WEIGHTS
from ..client import fetch_model
from ..utils import RESOLVE_FORMATS, ConversionContext, check_ext
@ -112,7 +112,7 @@ def convert_diffusion_diffusers_xl(
unet_path,
save_as_external_data=True,
all_tensors_to_one_file=True,
location="weights.pb",
location=ONNX_WEIGHTS,
)
return (True, dest_path)