Stable Diffusion源码调试(一)
个人模型主页:https://liblib.ai/userpage/369b11c9952245e28ea8d107ed9c2746/model
Stable Diffusion版本:https://github.com/AUTOMATIC1111/stable-diffusion-webui/releases/tag/v1.4.1
调试txt2img的参数和工作流
文件 /work/stable-diffusion-webui/modules/txt2img.py

进入txt2img函数
StableDiffusionProcessingTxt2Img和StableDiffusionProcessingImg2Img 继承StableDiffusionProcessing类
先去看父类StableDiffusionProcessing,注意到其中一个函数 sample,这个函数需要子类重写。
sample函数也是生图过程中,重要的函数处理。

去看看StableDiffusionProcessingTxt2Img类中的sample函数如何实现

文件 /work/stable-diffusion-webui/modules/processing.py
进入txt2img函数,再到process_images函数,process_images_inner函数, 在process_images_inner函数中调用sample函数得到图像。
进入process_images_inner函数,看一下

p.sample函数,执行n次迭代,得到latent空间出的图,


程序执行到箭头位置,可以知道,我们已经获得生成的图了。
往下执行就是 其他操作。







![[动态规划] (十一) 简单多状态 LeetCode 面试题17.16.按摩师 和 198.打家劫舍](https://img-blog.csdnimg.cn/img_convert/9ff88a675db2d5d9a0f0043b1a9c35af.png)











