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?

まず、exampleフォルダーのトレーニングスクリプトとBK-SDM蒸留モデルを使ってエラーを再現します。報告されている「unknown mid_block_type」失敗を、変更されたUNet構造と併せて確認し、そのうえでControlNetのトレーニングがこのモデルをサポートできるかどうかを判断します。完了の条件は、互換性の結果が文書化されていること、または動作するトレーニングパスがあることです。

索引モデルが issue の本文から書いたものです。