Stable_diffusion 2.1 源码详解——Unet部分
图中红框是源码中emb的生成的地方,emb是Unet中Resblock的输入,首先timesteps的初始化形状为(2,),经过timestep_embedding后生成timestep对应的embedding,形状为(2,320),在经过self.time_embed后,变成形状为(2,320)的输入。最后通过proj_out输出。每一次内循环结构的模块现在看看每一次内循环的实现,我们一边看下面的图一边对应代码,每一次内循环对应的是蓝框里的内容,每一次外循环对应的是红框里的内容。
原创
2024-09-05 16:18:34 ·
1160 阅读 ·
0 评论