Update README.md
Browse files
README.md
CHANGED
|
@@ -37,10 +37,10 @@ It is trained to act as a ControlNet module and requires a compatible SD base mo
|
|
| 37 |
- **Inputs:**
|
| 38 |
- `conditioning_image`: black-and-white manga scan (RGB)
|
| 39 |
- `text prompt`: optional (e.g. "colorized version of this panel")
|
| 40 |
-
- **Loss:** MSE with FP16, trained on 1×
|
| 41 |
- **Resolution:** 512x512
|
| 42 |
- **Scheduler:** default diffusers setup
|
| 43 |
-
- **Optimizer:**
|
| 44 |
|
| 45 |
---
|
| 46 |
|
|
|
|
| 37 |
- **Inputs:**
|
| 38 |
- `conditioning_image`: black-and-white manga scan (RGB)
|
| 39 |
- `text prompt`: optional (e.g. "colorized version of this panel")
|
| 40 |
+
- **Loss:** MSE with FP16, trained on 1×RTX3090, 4 epochs
|
| 41 |
- **Resolution:** 512x512
|
| 42 |
- **Scheduler:** default diffusers setup
|
| 43 |
+
- **Optimizer:** LR: `1.4e-4`
|
| 44 |
|
| 45 |
---
|
| 46 |
|