feat(api): create model path if needed during convert script
This commit is contained in:
parent
c34ddacf55
commit
6c22cb1cbf
|
@ -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)
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue