Stable Diffusion webui 源码调试(一)

个人模型主页:LibLibai
stable-diffusion-webui 版本: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空间出的图,

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

AIGC群交流

在这里插入图片描述

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐