Stable Diffusion webui 源码调试(二)

Stable Diffusion webui 源码调试(二)

个人模型主页:LibLibai

stable-diffusion-webui 版本:v1.4.1

内容更新随机,看心情调试代码~

分析StableDiffusionProcessingTxt2Img类中的sample函数

Sampler

/work/stable-diffusion-webui/modules/sd_samplers_kdiffusion.py

这个Sampler就是真正去执行生成 latent空间的图像了。

sample函数,接收了生图时需要的全部参数

可以看到初始latent图像就是 前面生成的随机图像x

最后就是self.launch_sampling函数的封装,进入到sample_euler_ancestral函数,生图,latent空间中的生图细节就在这个函数里面了。

这里我们用了euler采样器,换其他采样器同理。

这个函数返回x, 就是latent空间的图像,出去之后,调用decode部分的网络,就得到我们所看到的图像。

不执行数据增强的情况下,直接返回 latent空间的图像。

回到process_images_inner函数(/work/stable-diffusion-webui/modules/processing.py)

可以看到samples_ddim(latent空间的图像)经过decode_first_stage函数处理后,得到x,就是下图。

先整理到这里。

相关推荐
Icoolkj1 天前
深入了解 Stable Diffusion:AI 图像生成的奥秘
人工智能·stable diffusion
这是一个懒人1 天前
mac 快速安装stable diffusion webui
macos·stable diffusion
璇转的鱼2 天前
Stable Diffusion进阶之Controlnet插件使用
人工智能·ai作画·stable diffusion·aigc·ai绘画
AloneCat20123 天前
stable Diffusion模型结构
stable diffusion
西西弗Sisyphus3 天前
Stable Diffusion XL 文生图
stable diffusion
霍志杰4 天前
stable-diffusion windows本地部署
windows·stable diffusion
昨日之日20064 天前
ACE-Step - 20秒生成4分钟完整歌曲,音乐界的Stable Diffusion,支持50系显卡 本地一键整合包下载
计算机视觉·stable diffusion·音视频
白熊1885 天前
【图像大模型】Stable Diffusion Web UI:深度解析与实战指南
ui·stable diffusion
西西弗Sisyphus6 天前
基于Stable Diffusion XL模型进行文本生成图像的训练
stable diffusion
阿维的博客日记7 天前
VAE和Stable Diffusion的关系
stable diffusion·vae