From a552a6064cc8dca93b6eb5f7d6cefcd1257297ff Mon Sep 17 00:00:00 2001 From: blepping Date: Sat, 17 Feb 2024 15:15:01 -0700 Subject: [PATCH] Fix perlin noise for Stable Cascade --- extra_samplers.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/extra_samplers.py b/extra_samplers.py index a233019..c1acd57 100644 --- a/extra_samplers.py +++ b/extra_samplers.py @@ -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 @@ -223,7 +223,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() @@ -794,4 +794,4 @@ discard_penultimate_sigma_samplers = set(( "clyb_4m_sde_momentumized" )) -extra_schedulers = {} \ No newline at end of file +extra_schedulers = {}