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,就是下图。

先整理到这里。

相关推荐
春末的南方城市7 小时前
港科大&快手提出统一上下文视频编辑 UNIC,各种视频编辑任务一网打尽,还可进行多项任务组合!
人工智能·计算机视觉·stable diffusion·aigc·transformer
多恩Stone4 天前
【Stable Diffusion 1.5 】在 Unet 中每个 Cross Attention 块中的张量变化过程
stable diffusion
今夕节度使4 天前
ARM架构推理Stable Diffusiond
stable diffusion
远瞻。8 天前
【论文精读】2024 ECCV--MGLD-VSR现实世界视频超分辨率(RealWorld VSR)
人工智能·算法·stable diffusion·音视频·超分辨率重建
远瞻。9 天前
【论文精读】2024 CVPR--Upscale-A-Video现实世界视频超分辨率(RealWorld VSR)
论文阅读·人工智能·算法·stable diffusion·音视频·超分辨率重建
乱世刀疤10 天前
AI绘画:手把手带你Stable Diffusion从入门到精通(系列教程)
人工智能·ai作画·stable diffusion
layneyao11 天前
从0到1搭建AI绘画模型:Stable Diffusion微调全流程避坑指南
ai作画·stable diffusion
远瞻。11 天前
【论文精读】2024 arXiv --VEnhancer现实世界视频超分辨率(RealWorld VSR)
论文阅读·stable diffusion·音视频·超分辨率重建
立秋678914 天前
深入理解Diffusers: 从基础到Stable Diffusion
stable diffusion
Liudef0614 天前
Stable Diffusion底模对应的VAE推荐
stable diffusion