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