Instructions to use 8BitStudio/Aniimage-2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use 8BitStudio/Aniimage-2 with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("8BitStudio/Aniimage-2", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- Draw Things
- DiffusionBee
Update README.md
Browse files
README.md
CHANGED
|
@@ -31,7 +31,7 @@ It is not based on any existing models, the UNet is trained from scratch.
|
|
| 31 |
| **Dataset** | ~1.2 million curated anime images from Danbooru |
|
| 32 |
| **GPU** | NVIDIA RTX 5060 Ti 16GB |
|
| 33 |
| **Scheduler** | DPM++ SDE |
|
| 34 |
-
| **
|
| 35 |
|
| 36 |
## Requirements
|
| 37 |
|
|
@@ -142,10 +142,10 @@ The model generates male characters less reliably than female characters, it may
|
|
| 142 |
|
| 143 |
## What's Next
|
| 144 |
|
| 145 |
-
Aniimage-
|
| 146 |
The model will also have significantly more unique images.
|
| 147 |
Aniimage-2 used incorrect Min-SNR loss weighting with its v-prediction objective, causing near-pure-noise timesteps to receive excessive emphasis. This has been corrected for Aniimage-3.
|
| 148 |
-
Aniimage-
|
| 149 |
|
| 150 |
|
| 151 |
## License
|
|
|
|
| 31 |
| **Dataset** | ~1.2 million curated anime images from Danbooru |
|
| 32 |
| **GPU** | NVIDIA RTX 5060 Ti 16GB |
|
| 33 |
| **Scheduler** | DPM++ SDE |
|
| 34 |
+
| **Recommended steps** | 50|
|
| 35 |
|
| 36 |
## Requirements
|
| 37 |
|
|
|
|
| 142 |
|
| 143 |
## What's Next
|
| 144 |
|
| 145 |
+
Aniimage-2.5 is currently in the works and will have a much higher parameter count: 738.1M parameters compared to Aniimage-2 and Aniimage-1's 433.4M parameters.
|
| 146 |
The model will also have significantly more unique images.
|
| 147 |
Aniimage-2 used incorrect Min-SNR loss weighting with its v-prediction objective, causing near-pure-noise timesteps to receive excessive emphasis. This has been corrected for Aniimage-3.
|
| 148 |
+
Aniimage-2.5 also uses a much higher-quality dataset with improved filtering. More than 100,000 low-quality images from the Aniimage-2 dataset were removed.
|
| 149 |
|
| 150 |
|
| 151 |
## License
|