fix(api): unload UNet before CNet conversion
This commit is contained in:
parent
054e5c406a
commit
9ffe266384
|
@ -878,6 +878,9 @@ def convert_diffusion_diffusers_optimum(
|
|||
location=ONNX_WEIGHTS,
|
||||
)
|
||||
|
||||
del unet
|
||||
run_gc()
|
||||
|
||||
if conversion.control:
|
||||
logger.debug("converting CNet from pretrained UNet")
|
||||
cnet_path = convert_diffusion_diffusers_cnet(
|
||||
|
|
Loading…
Reference in New Issue