File size: 1,824 Bytes
cb82a2c
 
 
 
 
 
 
 
4d1f47b
c1113c9
cb82a2c
c99739e
cb82a2c
6a8056f
 
 
cb82a2c
8b6395e
 
 
 
0d33d8e
 
 
 
 
4966b84
8b6395e
0d33d8e
 
0443d1e
 
 
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
---
license: creativeml-openrail-m
tags:
- text-to-image
---
### 2d-art-sprites Dreambooth model trained by ana-tamais with [TheLastBen's fast-DreamBooth](https://colab.research.google.com/github/TheLastBen/fast-stable-diffusion/blob/main/fast-DreamBooth.ipynb) notebook



You can test this model using this [Colab Notebook for Inference](https://colab.research.google.com/drive/1pFaEJHa7mxFruBfm2hDnR8S6aEo7sAFx?usp=sharing)

Sample pictures of 2dart concept:

<img src="https://huggingface.co/ana-tamais/2d-art-sprites/resolve/main/concept_images/2dart_1.png" width=256></img>
<img src="https://huggingface.co/ana-tamais/2d-art-sprites/resolve/main/concept_images/2dart_4.png" width=256></img>
<img src="https://huggingface.co/ana-tamais/2d-art-sprites/resolve/main/concept_images/2dart_9.png" width=256></img>
    
We saved the training data in `dataset.zip`, and some generated results in `results.zip`.

### Some recommendations 
We recommend to set the:
- prompt as: `"[some wizard, paladin, healer, etc.], in the style of 2dart, white background, no background, full body"`
- negative prompt as: `"deformed, mutilated limbs, background, multiple people"`
- guidance scale between `9 and 10`
- sampling method as `Euler`
- sampling steps as `60`
- batch size as `1` - avoid high batch size, unless you have a high memory GPU

This set of hyperparameters lead us to stable and good results.

This model was trained using images which the character has some human format or is a not deformed living being. So if you try to predict something like "sword, mirror, candle, etc" (non-living things), we saw the model doesn't perform so well.

You need at least a Tesla T4 to be able to run the inference step using the given [notebook](https://colab.research.google.com/drive/1pFaEJHa7mxFruBfm2hDnR8S6aEo7sAFx?usp=sharing).