diff options
| author | Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> | 2023-08-14 13:34:51 +0800 |
|---|---|---|
| committer | Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> | 2023-08-14 13:34:51 +0800 |
| commit | e7c03ccdcefb2a80129703931ef1f8455708945b (patch) | |
| tree | 6769bdb3a32bdf99cf25422206ff03457bdfa86c /modules/mac_specific.py | |
| parent | d9cc27cb29926c9cc5dce331da8fbaf996cf4973 (diff) | |
| parent | 007ecfbb29771aa7cdcf0263ab1811bc75fa5446 (diff) | |
Merge branch 'dev' into extra-norm-module
Diffstat (limited to 'modules/mac_specific.py')
| -rw-r--r-- | modules/mac_specific.py | 3 |
1 files changed, 0 insertions, 3 deletions
diff --git a/modules/mac_specific.py b/modules/mac_specific.py index bce527cc..89256c5b 100644 --- a/modules/mac_specific.py +++ b/modules/mac_specific.py @@ -52,9 +52,6 @@ def cumsum_fix(input, cumsum_func, *args, **kwargs): if has_mps: - # MPS fix for randn in torchsde - CondFunc('torchsde._brownian.brownian_interval._randn', lambda _, size, dtype, device, seed: torch.randn(size, dtype=dtype, device=torch.device("cpu"), generator=torch.Generator(torch.device("cpu")).manual_seed(int(seed))).to(device), lambda _, size, dtype, device, seed: device.type == 'mps') - if platform.mac_ver()[0].startswith("13.2."): # MPS workaround for https://github.com/pytorch/pytorch/issues/95188, thanks to danieldk (https://github.com/explosion/curated-transformers/pull/124) CondFunc('torch.nn.functional.linear', lambda _, input, weight, bias: (torch.matmul(input, weight.t()) + bias) if bias is not None else torch.matmul(input, weight.t()), lambda _, input, weight, bias: input.numel() > 10485760) |