diff options
author | Volpeon <git@volpeon.ink> | 2023-06-21 21:46:15 +0200 |
---|---|---|
committer | Volpeon <git@volpeon.ink> | 2023-06-21 21:46:15 +0200 |
commit | b5d3df18c3a56699a3658ad58a02d4494836972f (patch) | |
tree | 8a43468111eee827564bb5d1561d2d4910915c61 /training/functional.py | |
parent | Update (diff) | |
download | textual-inversion-diff-b5d3df18c3a56699a3658ad58a02d4494836972f.tar.gz textual-inversion-diff-b5d3df18c3a56699a3658ad58a02d4494836972f.tar.bz2 textual-inversion-diff-b5d3df18c3a56699a3658ad58a02d4494836972f.zip |
Update
Diffstat (limited to 'training/functional.py')
-rw-r--r-- | training/functional.py | 13 |
1 files changed, 0 insertions, 13 deletions
diff --git a/training/functional.py b/training/functional.py index f68faf9..3c7848f 100644 --- a/training/functional.py +++ b/training/functional.py | |||
@@ -348,7 +348,6 @@ def loss_step( | |||
348 | guidance_scale: float, | 348 | guidance_scale: float, |
349 | prior_loss_weight: float, | 349 | prior_loss_weight: float, |
350 | seed: int, | 350 | seed: int, |
351 | offset_noise_strength: float, | ||
352 | input_pertubation: float, | 351 | input_pertubation: float, |
353 | disc: Optional[ConvNeXtDiscriminator], | 352 | disc: Optional[ConvNeXtDiscriminator], |
354 | min_snr_gamma: int, | 353 | min_snr_gamma: int, |
@@ -377,16 +376,6 @@ def loss_step( | |||
377 | ) | 376 | ) |
378 | applied_noise = noise | 377 | applied_noise = noise |
379 | 378 | ||
380 | if offset_noise_strength != 0: | ||
381 | applied_noise = applied_noise + offset_noise_strength * perlin_noise( | ||
382 | latents.shape, | ||
383 | res=1, | ||
384 | octaves=4, | ||
385 | dtype=latents.dtype, | ||
386 | device=latents.device, | ||
387 | generator=generator, | ||
388 | ) | ||
389 | |||
390 | if input_pertubation != 0: | 379 | if input_pertubation != 0: |
391 | applied_noise = applied_noise + input_pertubation * torch.randn( | 380 | applied_noise = applied_noise + input_pertubation * torch.randn( |
392 | latents.shape, | 381 | latents.shape, |
@@ -751,7 +740,6 @@ def train( | |||
751 | global_step_offset: int = 0, | 740 | global_step_offset: int = 0, |
752 | guidance_scale: float = 0.0, | 741 | guidance_scale: float = 0.0, |
753 | prior_loss_weight: float = 1.0, | 742 | prior_loss_weight: float = 1.0, |
754 | offset_noise_strength: float = 0.01, | ||
755 | input_pertubation: float = 0.1, | 743 | input_pertubation: float = 0.1, |
756 | disc: Optional[ConvNeXtDiscriminator] = None, | 744 | disc: Optional[ConvNeXtDiscriminator] = None, |
757 | schedule_sampler: Optional[ScheduleSampler] = None, | 745 | schedule_sampler: Optional[ScheduleSampler] = None, |
@@ -814,7 +802,6 @@ def train( | |||
814 | guidance_scale, | 802 | guidance_scale, |
815 | prior_loss_weight, | 803 | prior_loss_weight, |
816 | seed, | 804 | seed, |
817 | offset_noise_strength, | ||
818 | input_pertubation, | 805 | input_pertubation, |
819 | disc, | 806 | disc, |
820 | min_snr_gamma, | 807 | min_snr_gamma, |