Merge branch 'fix_perlin_cascade'

2024-03-04 12:35:58 -06:00
parent 1f161b20f9 a552a6064c
commit 76d98c3b84
1 changed files with 3 additions and 3 deletions
@@ -1,7 +1,7 @@
 import math

 import torch
-from torch import nn
+from torch import nn, FloatTensor
 import torchsde
 from tqdm.auto import trange, tqdm

@@ -225,7 +225,7 @@ def rand_perlin_like(x):
    noise_size_W = noise.size(dim=3)
    perlin = None
    for i in range(2):
-        noise += perlin_noise((noise_size_H, noise_size_W), (noise_size_H, noise_size_W), batch_size=4).to(x.device)
+        noise += perlin_noise((noise_size_H, noise_size_W), (noise_size_H, noise_size_W), batch_size=x.shape[1]).to(x.device)
    #noise += perlin
    #print(noise)
    return noise / noise.std()
@@ -829,4 +829,4 @@ def get_sigmas_simple_exponential(model, steps):

 extra_schedulers = {
    "simple_exponential": get_sigmas_simple_exponential
-}
+}