Merge pull request #1089 from ilikenwf/vladpull1

Add missing check for empty tensor
This commit is contained in:
Vladimir Mandic
2023-05-20 08:29:11 -04:00
committed by GitHub
+9 -1
View File
@@ -308,7 +308,15 @@ class KDiffusionSampler:
def create_noise_sampler(self, x, sigmas, p):
from k_diffusion.sampling import BrownianTreeNoiseSampler
sigma_min, sigma_max = sigmas[sigmas > 0].min(), sigmas.max()
positive_sigmas = sigmas[sigmas > 0]
if positive_sigmas.numel() > 0:
sigma_min = positive_sigmas.min(dim=0)[0]
else:
sigma_min = 0
sigma_max = sigmas.max()
current_iter_seeds = p.all_seeds[p.iteration * p.batch_size:(p.iteration + 1) * p.batch_size]
return BrownianTreeNoiseSampler(x, sigma_min, sigma_max, seed=current_iter_seeds)