Add sample_start_step setting to configure when sampling starts (#949)

Co-authored-by: Jaret Burkett <jaretburkett@gmail.com>
This commit is contained in:
fatalisandJaret Burkett authored and GitHub committed 2026-07-15 11:15:50 -06:00
1 parent 4ece17b71f
commit 8bbd051667
31 files changed
+46 -2

No files matched your search

+1
View File
@@ -80,6 +80,7 @@ class SampleConfig:
def __init__(self, **kwargs):
self.sampler: str = kwargs.get('sampler', 'ddpm')
self.sample_every: int = kwargs.get('sample_every', 100)
self.sample_start_step: int = kwargs.get('sample_start_step', 0)
self.width: int = kwargs.get('width', 512)
self.height: int = kwargs.get('height', 512)
self.neg = kwargs.get('neg', False)