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

Scale sigmoid to default 1.0

Browse Source
This commit is contained in:
rockerBOO
2025-03-20 15:09:11 -04:00
parent e8b3254858
commit 8aa126582e
3 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
pytest.ini
View File

@@ -6,3 +6,4 @@ filterwarnings =
ignore::DeprecationWarning ignore::DeprecationWarning
ignore::UserWarning ignore::UserWarning
ignore::FutureWarning ignore::FutureWarning
pythonpath = .

2
requirements.txt
View File

@@ -7,7 +7,7 @@ opencv-python==4.8.1.78
einops==0.7.0 einops==0.7.0
pytorch-lightning==1.9.0 pytorch-lightning==1.9.0
bitsandbytes==0.44.0 bitsandbytes==0.44.0
prodigyopt==1.0 prodigyopt>=1.0
lion-pytorch==0.0.6 lion-pytorch==0.0.6
schedulefree==1.4 schedulefree==1.4
tensorboard tensorboard

4
tests/library/test_flux_train_utils.py
View File

@@ -77,7 +77,7 @@ def test_uniform_sampling(args, noise_scheduler, latents, noise, device):
def test_sigmoid_sampling(args, noise_scheduler, latents, noise, device): def test_sigmoid_sampling(args, noise_scheduler, latents, noise, device):
args.timestep_sampling = "sigmoid" args.timestep_sampling = "sigmoid"
args.sigmoid_scale = 10.0 args.sigmoid_scale = 1.0
dtype = torch.float32 dtype = torch.float32
noisy_input, timesteps, sigmas = get_noisy_model_input_and_timesteps(args, noise_scheduler, latents, noise, device, dtype) noisy_input, timesteps, sigmas = get_noisy_model_input_and_timesteps(args, noise_scheduler, latents, noise, device, dtype)
@@ -102,7 +102,7 @@ def test_shift_sampling(args, noise_scheduler, latents, noise, device):
def test_flux_shift_sampling(args, noise_scheduler, latents, noise, device): def test_flux_shift_sampling(args, noise_scheduler, latents, noise, device):
args.timestep_sampling = "flux_shift" args.timestep_sampling = "flux_shift"
args.sigmoid_scale = 10.0 args.sigmoid_scale = 1.0
dtype = torch.float32 dtype = torch.float32
noisy_input, timesteps, sigmas = get_noisy_model_input_and_timesteps(args, noise_scheduler, latents, noise, device, dtype) noisy_input, timesteps, sigmas = get_noisy_model_input_and_timesteps(args, noise_scheduler, latents, noise, device, dtype)