apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

apply lint

This commit is contained in:
Sean Sube 2023-12-19 23:01:05 -06:00
parent c8a9734acf
commit e09631ef05
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/diffusion/diffusion.py
View File

@ -10,7 +10,7 @@
### ###
from logging import getLogger from logging import getLogger
from os import path, makedirs from os import makedirs, path
from pathlib import Path from pathlib import Path
from shutil import rmtree from shutil import rmtree
from typing import Any, Dict, Optional, Tuple, Union from typing import Any, Dict, Optional, Tuple, Union
@ -131,10 +131,14 @@ def convert_diffusion_diffusers_cnet(
makedirs(cnet_tmp, exist_ok=True) makedirs(cnet_tmp, exist_ok=True)
unet.save_pretrained(cnet_tmp) 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: else:
logger.debug("loading CNet from pretrained UNet config") 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) pipe_cnet = pipe_cnet.to(device=device, dtype=dtype)
run_gc() run_gc()