search

Nota-NetsPresso / BK-SDM

A Compressed Stable Diffusion for Efficient Text-to-Image Generation [ECCV'24]
Other
238 stars 16 forks source link

controlnet on bk-sdm-v2-tiny #64

Open coolKeen opened 1 month ago

coolKeen commented 1 month ago

Hi, thanks for your great job!!! I want to train controlnet using bk-sdm-v2-tiny. The controlnet training code if from https://github.com/huggingface/diffusers/tree/main/examples/controlnet. But When I run the cmd like below image I get the error " unknown mid_block_type : None" below image，Without mid_block， is there any way to make bk-sdm-v2-tiny support controlnet training? Thanks in advance!