Method generate_noise

diffsynth/pipelines/base.py:124–127

(self, shape, seed=None, device="cpu", dtype=torch.float16)

    
    def generate_noise(self, shape, seed=None, device="cpu", dtype=torch.float16):
        generator = None if seed is None else torch.Generator(device).manual_seed(seed)
        noise = torch.randn(shape, generator=generator, device=device, dtype=dtype)
        return noise