diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-05-17 20:18:56 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-05-17 20:18:56 +0000 |
| commit | 04b4508a66de58c9f3a422fdcad4dd2ec3ad39ce (patch) | |
| tree | 1cb30a63099a69f678d4901b495203b765a6df59 /modules/devices.py | |
| parent | 7201d940a4fe664beb9662fadbeade4ee1d788f7 (diff) | |
| parent | b397f63e00bbfbe9087d80abb457aa9a593b181b (diff) | |
| download | stable-diffusion-webui-gfx803-04b4508a66de58c9f3a422fdcad4dd2ec3ad39ce.tar.gz stable-diffusion-webui-gfx803-04b4508a66de58c9f3a422fdcad4dd2ec3ad39ce.tar.bz2 stable-diffusion-webui-gfx803-04b4508a66de58c9f3a422fdcad4dd2ec3ad39ce.zip | |
Merge branch 'dev' into improve-frontend-responsiveness
Diffstat (limited to 'modules/devices.py')
| -rw-r--r-- | modules/devices.py | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/modules/devices.py b/modules/devices.py index 52c3e7cd..d8a34a0f 100644 --- a/modules/devices.py +++ b/modules/devices.py @@ -65,7 +65,7 @@ def enable_tf32(): # enabling benchmark option seems to enable a range of cards to do fp16 when they otherwise can't # see https://github.com/AUTOMATIC1111/stable-diffusion-webui/pull/4407 - if any([torch.cuda.get_device_capability(devid) == (7, 5) for devid in range(0, torch.cuda.device_count())]): + if any(torch.cuda.get_device_capability(devid) == (7, 5) for devid in range(0, torch.cuda.device_count())): torch.backends.cudnn.benchmark = True torch.backends.cuda.matmul.allow_tf32 = True @@ -92,14 +92,18 @@ def cond_cast_float(input): def randn(seed, shape): + from modules.shared import opts + torch.manual_seed(seed) - if device.type == 'mps': + if opts.randn_source == "CPU" or device.type == 'mps': return torch.randn(shape, device=cpu).to(device) return torch.randn(shape, device=device) def randn_without_seed(shape): - if device.type == 'mps': + from modules.shared import opts + + if opts.randn_source == "CPU" or device.type == 'mps': return torch.randn(shape, device=cpu).to(device) return torch.randn(shape, device=device) |