Edit model card

English Version

Example Fine-Tuned Model for learning diffusion model

My first fintuning model through python script

base model: google/ddpm-celebahq-256

fine-tuning dataset: vintageface

The loss is not very stable, and the output is not very satisfactory. But since it's a study case, you can't ask for too much.

image/png

image/png

Usage

from diffusers import DDPMPipeline

pipeline = DDPMPipeline.from_pretrained('Chilli-b/my-finetuned-model-celebahq-on-VintageFaces-4epochs')
image = pipeline().images[0]
image

中文版

学习扩散模型时训练的微调模型

这是我第一个微调模型,也是第一个通过 Python 脚本进行训练的微调模型。

基底模型:google/ddpm-celebahq-256

微调数据集:Norod78/Vintage-Faces-FFHQAligned

loss 不是很稳定,输出结果也不太令人满意。 不过既然是学习案例,那也不能要求太多。

image/png

image/png

模型使用

from diffusers import DDPMPipeline

pipeline = DDPMPipeline.from_pretrained('Chilli-b/my-finetuned-model-celebahq-on-VintageFaces-4epochs')
image = pipeline().images[0]
image
Downloads last month
58
Inference API does not yet support diffusers models for this pipeline type.

Dataset used to train Chilli-b/my-finetuned-model-celebahq-on-VintageFaces-4epochs