feat: fix randn found element of type float at pos 2

Signed-off-by: storyicon <storyicon@foxmail.com>
This commit is contained in:
storyicon 2023-11-17 09:48:18 +00:00
parent 4afaaf8a02
commit 4f2a4a3615

View File

@ -110,7 +110,7 @@ class ImageRNG:
self.is_first = True
def first(self):
noise_shape = self.shape if self.seed_resize_from_h <= 0 or self.seed_resize_from_w <= 0 else (self.shape[0], self.seed_resize_from_h // 8, self.seed_resize_from_w // 8)
noise_shape = self.shape if self.seed_resize_from_h <= 0 or self.seed_resize_from_w <= 0 else (self.shape[0], int(self.seed_resize_from_h) // 8, int(self.seed_resize_from_w // 8))
xs = []