furusu commited on
Commit
8f745bf
1 Parent(s): 507d082

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -1
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