mirror of
https://github.com/vladmandic/automatic
synced 2026-09-18 16:54:33 +02:00
Merge pull request #1089 from ilikenwf/vladpull1
Add missing check for empty tensor
This commit is contained in:
@@ -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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user