EEG to Image setup with diffusers

Hey, I am very new to diffusers package. I want a reference where I can just use the frozen UNet and the frozen decoder to which I can attach an EEG encoder and just train the new encoder. Are there any code examples that take apart the Stable diffusion components so that I can see how to implement this new setup. I tried looking at control net but that wasn’t what I was looking for.

1 Like