File size: 1,748 Bytes
e40d431 620b0fb e40d431 620b0fb e40d431 6fd3f35 620b0fb 6fd3f35 620b0fb e40d431 620b0fb 6fd3f35 620b0fb 6fd3f35 620b0fb 6fd3f35 620b0fb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
---
license: mit
tags:
- pytorch
- diffusers
- unconditional-image-generation
- diffusion-models-class
datasets:
- Norod78/Vintage-Faces-FFHQAligned
---
# 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.
### Usage
```python
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 不是很稳定，输出结果也不太令人满意。
> 不过既然是学习案例，那也不能要求太多。
### 模型使用
```python
from diffusers import DDPMPipeline
pipeline = DDPMPipeline.from_pretrained('Chilli-b/my-finetuned-model-celebahq-on-VintageFaces-4epochs')
image = pipeline().images[0]
image
``` |