Skip to content

Releases: owenliang/pytorch-diffusion

Release list

20230806

20230806 Latest
Latest

Choose a tag to compare

View all tags
@owenliang owenliang released this 06 Aug 06:53
20230806
bc8fbdc
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Verified
Learn about vigilant mode.

引入lora微调机制，训练用lora_finetune.py，预测继续沿用denoise.py

Assets 4
Loading

20230729

20230729

Choose a tag to compare

View all tags
@owenliang owenliang released this 29 Jul 14:40
20230729
0e09052
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Verified
Learn about vigilant mode.

在unet中引入attention机制，模型支持输入期望生成的数字，去噪过程将生成你想要的数字图片。

附件是已经预训练好的模型，放在根目录下运行denoise.py可以看到推理效果。

Loading

20230723

20230723

Choose a tag to compare

View all tags
@owenliang owenliang released this 23 Jul 00:41
20230723
ed18aba

显存不够的同学可以只玩推理，下载已经训练好的model.pt，放到项目根目录，运行denoise.py生成图像

Loading