VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

stratified_zero_thresholdに変更

Browse Source
This commit is contained in:
u-haru
2023-03-31 01:14:03 +09:00
parent 4dacc52bde
commit dade23a414
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
networks/lora.py
View File

@@ -202,7 +202,7 @@ def create_network(multiplier, network_dim, network_alpha, vae, text_encoder, un
if "," in down_weight: if "," in down_weight:
down_weight = [float(s) for s in down_weight.split(",") if s] down_weight = [float(s) for s in down_weight.split(",") if s]
network.set_stratified_lr_weight(up_weight,float(kwargs.get('mid_weight', 1.0)),down_weight,float(kwargs.get('lr_weight_threshold', 0.0))) network.set_stratified_lr_weight(up_weight,float(kwargs.get('mid_weight', 1.0)),down_weight,float(kwargs.get('stratified_zero_threshold', 0.0)))
return network return network