controlNet training for dstilled models

Is it possible to train controlNet for distilled model (Block-removed Knowledge-distilled Stable Diffusion, https://github.com/Nota-NetsPresso/BK-SDM)?

When I used scripts in example folder for distilled model, it returns error, similar to "unknown mid_block_type ...." , probably because of unet structure changed. Are there any ways to train controlNet for distilled model?

Empieza reproduciendo el error con los scripts de entrenamiento de la carpeta example y el modelo destilado BK-SDM. Lee el error notificado «unknown mid_block_type» junto con la estructura modificada de UNet y determina después si el entrenamiento de ControlNet puede admitir este modelo; se considera terminado cuando haya un resultado de compatibilidad documentado o una ruta de entrenamiento funcional.

Escrito por el modelo de indexación a partir del texto del issue.