apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): make sure new SD converter does not produce extra files

This commit is contained in:
Sean Sube 2023-12-24 10:47:46 -06:00
parent 29ccd16437
commit 15b9b51ed5
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -895,6 +895,8 @@ def convert_diffusion_diffusers_optimum(
v2=v2, v2=v2,
) )
collate_cnet(cnet_path)
if conversion.half: if conversion.half:
infer_shapes_path(cnet_path) infer_shapes_path(cnet_path)
cnet = load_model(cnet_path) cnet = load_model(cnet_path)