ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-29 18:51:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added ability to use consistent noise for each image in a dataset by hashing the path and using that as a seed.

Browse Source
This commit is contained in:
Jaret Burkett
2025-02-08 07:13:48 -07:00
parent af5e760be1
commit c6d8eedb94
3 changed files with 31 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
toolkit/config_modules.py
View File

@@ -407,6 +407,9 @@ class TrainConfig:
# optimal noise pairing
self.optimal_noise_pairing_samples = kwargs.get('optimal_noise_pairing_samples', 1)
# forces same noise for the same image at a given size.
self.force_consistent_noise = kwargs.get('force_consistent_noise', False)
class ModelConfig: