From e09631ef05e6b2ea5600a447648c18a66bb5c288 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Tue, 19 Dec 2023 23:01:05 -0600 Subject: [PATCH] apply lint --- api/onnx_web/convert/diffusion/diffusion.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/api/onnx_web/convert/diffusion/diffusion.py b/api/onnx_web/convert/diffusion/diffusion.py index b6f43e2c..1e082c5b 100644 --- a/api/onnx_web/convert/diffusion/diffusion.py +++ b/api/onnx_web/convert/diffusion/diffusion.py @@ -10,7 +10,7 @@ ### from logging import getLogger -from os import path, makedirs +from os import makedirs, path from pathlib import Path from shutil import rmtree from typing import Any, Dict, Optional, Tuple, Union @@ -131,10 +131,14 @@ def convert_diffusion_diffusers_cnet( makedirs(cnet_tmp, exist_ok=True) unet.save_pretrained(cnet_tmp) - pipe_cnet = UNet2DConditionModel_CNet.from_pretrained(cnet_tmp, low_cpu_mem_usage=False) + pipe_cnet = UNet2DConditionModel_CNet.from_pretrained( + cnet_tmp, low_cpu_mem_usage=False + ) else: logger.debug("loading CNet from pretrained UNet config") - pipe_cnet = UNet2DConditionModel_CNet.from_pretrained(source, subfolder="unet", low_cpu_mem_usage=False) + pipe_cnet = UNet2DConditionModel_CNet.from_pretrained( + source, subfolder="unet", low_cpu_mem_usage=False + ) pipe_cnet = pipe_cnet.to(device=device, dtype=dtype) run_gc()