apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

feat(api): create model path if needed during convert script

This commit is contained in:
Sean Sube 2023-01-31 17:09:23 -06:00
parent c34ddacf55
commit 6c22cb1cbf
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
api/onnx_web/convert.py
View File

@ -10,7 +10,7 @@ from diffusers import (
)
from logging import getLogger
from onnx import load, save_model
from os import mkdir, path, environ
from os import environ, makedirs, mkdir, path
from pathlib import Path
from shutil import copyfile, rmtree
from sys import exit
@ -253,7 +253,7 @@ def convert_diffuser(name: str, url: str, opset: int, half: bool, token: str, si
)
del pipeline.text_encoder
logger.info('UNET config: %s', pipeline.unet.config)
logger.debug('UNET config: %s', pipeline.unet.config)
# UNET
if single_vae:
@ -307,7 +307,7 @@ def convert_diffuser(name: str, url: str, opset: int, half: bool, token: str, si
del pipeline.unet
if single_vae:
logger.info('VAE config: %s', pipeline.vae.config)
logger.debug('VAE config: %s', pipeline.vae.config)
# SINGLE VAE
vae_only = pipeline.vae
@ -517,6 +517,10 @@ def main() -> int:
args = parser.parse_args()
logger.info('CLI arguments: %s', args)
if path.exists(model_path):
logger.info('Model path does not existing, creating: %s', model_path)
makedirs(model_path)
logger.info('Converting base models.')
load_models(args, base_models)