Update README.md
Browse files
README.md
CHANGED
|
@@ -9,10 +9,17 @@ Put the yaml file with model because it is sd2.1 v_prediction model.
|
|
| 9 |
Model is the difference from the original checkpoint created by [extract_controlnet_diff.py](https://github.com/Mikubill/sd-webui-controlnet/blob/main/extract_controlnet_diff.py).
|
| 10 |
|
| 11 |
# diff_control_wd15beta2_canny.safetensors
|
| 12 |
-
|
| 13 |
+ target model: https://huggingface.co/waifu-diffusion/wd-1-5-beta2
|
| 14 |
+ dataset: 11k 1girl images
|
| 15 |
+ epoch: 2
|
| 16 |
+ batch size: 16
|
| 17 |
+ lr: 5e-5
|
| 18 |
+ amp:bfloat16
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
Model is the difference from the original checkpoint created by [extract_controlnet_diff.py](https://github.com/Mikubill/sd-webui-controlnet/blob/main/extract_controlnet_diff.py).
|
| 10 |
|
| 11 |
# diff_control_wd15beta2_canny.safetensors
|
|
|
|
| 12 |
+ target model: https://huggingface.co/waifu-diffusion/wd-1-5-beta2
|
| 13 |
+ dataset: 11k 1girl images
|
| 14 |
+ epoch: 2
|
| 15 |
+ batch size: 16
|
| 16 |
+ lr: 5e-5
|
| 17 |
+ amp:bfloat16
|
| 18 |
+
|
| 19 |
+
# diff_control_wd15beta2_pose.safetensors
|
| 20 |
+
+ target model: https://huggingface.co/waifu-diffusion/wd-1-5-beta2
|
| 21 |
+
+ dataset: 14k umamusume images. **Therefore, the model may have a bias**
|
| 22 |
+
+ epoch: 5 (The “sudden convergence phenomenon” occurred at the fourth epoch)
|
| 23 |
+
+ batch size: 16
|
| 24 |
+
+ lr: 5e-5
|
| 25 |
+
+ amp:bfloat16
|