Diffusion剪枝项目

参考BK-SDM(宝藏!)

Nota-NetsPresso/BK-SDM

参考数据集

https://huggingface.co/datasets/recastai/LAION-art-EN-improved-captions

PASD剪枝工程实践

graph TB
	s(PASD端侧落地) 
	
	s --> s0(基于BK-SDM预训练的1.5模型)
	s --> s1(基于BK-SDM\n预训练的SDXL模型)

	s0 --> s00(暂定基于segmind/tiny-sd) --> s000(推理跑起来)
	s000 --> s0000(移植到PASD训练代码\n开始训练PASD) --> s00000(进行LCM蒸馏) --> s000000(看实际效果)
	s000 --> s0001(直接基于该模型进行LCM蒸馏) --> s00010(看效果衰减)

基于BK-SDM预训练的1.5模型

基础的1.5模型: SG161222/Realistic_Vision_V4.0_noVAE · Hugging Face 剪枝的模型:segmind/tiny-sd · Hugging Face 基于上述模型进行LCM蒸馏的结果: