feat: fix randn found element of type float at pos 2
Signed-off-by: storyicon <storyicon@foxmail.com>
This commit is contained in:
parent
4afaaf8a02
commit
4f2a4a3615
|
@ -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 = []
|
||||
|
||||
|
|
Loading…
Reference in New Issue