ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-01-26 16:39:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added ability to pair samples with a closer noise with optimal_noise_pairing_samples

Browse Source
This commit is contained in:
Jaret Burkett
2025-01-21 18:30:10 -07:00
parent 29122b1a54
commit 89dd041b97
2 changed files with 32 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
toolkit/config_modules.py
View File

@@ -403,6 +403,9 @@ class TrainConfig:
# diffusion feature extractor
self.diffusion_feature_extractor_path = kwargs.get('diffusion_feature_extractor_path', None)
# optimal noise pairing
self.optimal_noise_pairing_samples = kwargs.get('optimal_noise_pairing_samples', 1)
class ModelConfig: