ControlNet / README.md
furusu's picture
Update README.md
ee652e9
|
raw
history blame
No virus
1.77 kB
---
license: openrail
tags:
- stable diffusion
- controlnet
---
Can be used in [sd-webui-controlnet](https://github.com/Mikubill/sd-webui-controlnet). Put the yaml file with model because it is sd2.1 v_prediction model.
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).
These are not the final version and need to be adjusted to be more accurate. Someone please do this.
# wd15_controlnet_edge_depth_pose.ckpt
+ target model: https://huggingface.co/waifu-diffusion/wd-1-5-beta2
+ dataset: 20k 1girl images
+ epoch: 5
+ batch size: 8
+ lr: 5e-5
+ amp:bfloat16
# diff_control_wd15beta2_canny.safetensors
**updated in 2023/03/05.**
+ target model: https://huggingface.co/waifu-diffusion/wd-1-5-beta2
+ dataset: 11k 1girl images
+ epoch: 2
+ batch size: 16
+ lr: 5e-5
+ amp:bfloat16
and resumed on:
+ dataset: 60k 1girl images
+ epoch: 2
+ batch size: 16
+ lr: 1e-5
+ amp:bfloat16
# diff_control_wd15beta2_pose.safetensors
**updated in 2023/03/06.**
+ target model: https://huggingface.co/waifu-diffusion/wd-1-5-beta2
+ dataset: 14k umamusume images. **Therefore, the model may have a bias**
+ epoch: 5 (The “sudden convergence phenomenon” occurred at the fourth epoch)
+ batch size: 16
+ lr: 5e-5
+ amp:bfloat16
and resumed on:
+ dataset: 52k 1girl images
+ epoch: 2
+ batch size: 16
+ lr: 1e-5
+ amp:bfloat16
# diff_control_wd15beta2_depth.safetensors
**updated in 2023/03/06.**
+ target model: https://huggingface.co/waifu-diffusion/wd-1-5-beta2
+ dataset: 11k 1girl images
+ epoch: 2
+ batch size: 16
+ lr: 5e-5
+ amp:bfloat16
and resumed on:
+ dataset: 60k 1girl images
+ epoch: 1
+ batch size: 16
+ lr: 1e-5
+ amp:bfloat16