diff --git a/torchsde/_brownian/brownian_interval.py b/torchsde/_brownian/brownian_interval.py index 79391d2..b9fa036 100644 --- a/torchsde/_brownian/brownian_interval.py +++ b/torchsde/_brownian/brownian_interval.py @@ -28,7 +28,7 @@ def _randn(size, dtype, device, seed): - generator = torch.Generator(device).manual_seed(int(seed)) + generator = torch.Generator().manual_seed(int(seed)) return torch.randn(size, dtype=dtype, device=device, generator=generator)