miniDiffusion,一个用PyTorch重新实现的Stable Diffusion 3.5

miniDiffusion,一个用PyTorch重新实现的Stable Diffusion 3.5模型的项目。它以最少的代码(约2800行)从头开始复现Stable Diffusion 3.5,包括从VAE到DiT以及训练和数据集脚本。项目包含核心图像生成模块、文本编码器、字节对和单体分词器等组件,还提供训练和推理脚本。该项目用于教育和实验目的。

参考文献:
[1] http://github.com/yousef-rafat/miniDiffusion



知识星球服务内容:Dify源码剖析及答疑,Dify对话系统源码,NLP电子书籍报告下载,公众号所有付费资料。加微信buxingtianxia21进NLP工程化资料群

(文:NLP工程化)

发表评论