Update README.md
Browse files
README.md
CHANGED
|
@@ -32,15 +32,10 @@ Aniimage-1 has been trained on 830,001 anime images from [Danbooru](https://danb
|
|
| 32 |
| **GPU** | NVIDIA RTX 5060 Ti 16GB |
|
| 33 |
| **Scheduler** | DDIM or DPM ++ 2M |
|
| 34 |
|
| 35 |
-
##
|
| 36 |
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
gen = Generator()
|
| 41 |
-
gen.load_model("path/to/checkpoint")
|
| 42 |
-
image = gen.generate(prompt="a smiling anime girl with long blue hair", steps=50, cfg_scale=7.0)
|
| 43 |
-
```
|
| 44 |
|
| 45 |
## Quick Start
|
| 46 |
|
|
|
|
| 32 |
| **GPU** | NVIDIA RTX 5060 Ti 16GB |
|
| 33 |
| **Scheduler** | DDIM or DPM ++ 2M |
|
| 34 |
|
| 35 |
+
## Requirements
|
| 36 |
|
| 37 |
+
- **GPU**: ~3.4 GB VRAM minimum (recommend 4+ GB)
|
| 38 |
+
- **CPU**: ~2 GB RAM. Image generation is extremely slow on cpu.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 39 |
|
| 40 |
## Quick Start
|
| 41 |
|