SubMaroon commited on
Commit
b0bca83
·
verified ·
1 Parent(s): 33bfe63

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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×A100, 20 epochs
41
  - **Resolution:** 512x512
42
  - **Scheduler:** default diffusers setup
43
- - **Optimizer:** AdamW, LR: `1e-5`
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