zwh19960817 / pytorch-diffusion-note

pytorch复现stable diffusion

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

pytorch-diffusion

基于mnist手写数字数据集的stable diffusion模型复现

可以输入引导数字,生成对应数字的图片

模型效果

引导生成10张图片,分别对应数字0~9

模型效果

训练主模型

python train.py

Lora微调

python lora_finetune.py

About

pytorch复现stable diffusion


Languages

Language:Python 100.0%