Assume 0 = inf for s_tmax

This commit is contained in:
catboxanon 2023-08-05 23:50:33 -04:00
parent ce4be668fe
commit 8f31b139b8

View File

@ -347,7 +347,7 @@ class KDiffusionSampler:
if len(self.extra_params) > 0: if len(self.extra_params) > 0:
s_churn = getattr(opts, 's_churn', p.s_churn) s_churn = getattr(opts, 's_churn', p.s_churn)
s_tmin = getattr(opts, 's_tmin', p.s_tmin) s_tmin = getattr(opts, 's_tmin', p.s_tmin)
s_tmax = getattr(opts, 's_tmax', p.s_tmax) s_tmax = getattr(opts, 's_tmax', p.s_tmax) or self.s_tmax # 0 = inf
s_noise = getattr(opts, 's_noise', p.s_noise) s_noise = getattr(opts, 's_noise', p.s_noise)
if s_churn != self.s_churn: if s_churn != self.s_churn: