how to train train new model with different channel number in unet

CompVis / stable-diffusion

A latent text-to-image diffusion model

https://ommer-lab.com/research/latent-diffusion-models/

Other

66.51k stars 9.97k forks source link

how to train train new model with different channel number in unet #833

Open jonathanyang0227 opened 4 months ago

jonathanyang0227 commented 4 months ago

currently the config of unet channel is : "block_out_channels": [320,640,1280,1280], i want to change the channel number, do i need to train it from scratch? how to modified it thanks for your help

happypic commented 4 months ago

![Uploading 1b1703bc9730444ad5ebd03686a47ea9.png…]()

jonathanyang0227 commented 4 months ago

@happypic hi i can't see your pic. Can you upload it again?