diff --git a/modules/util.py b/modules/util.py index 7958803..a8ca234 100644 --- a/modules/util.py +++ b/modules/util.py @@ -250,7 +250,8 @@ def __init__(self, channels, scale): [ torch.arange(size, dtype=torch.float32) for size in kernel_size - ] + ], + indexing="ij" ) for size, std, mgrid in zip(kernel_size, sigma, meshgrids): mean = (size - 1) / 2