hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 11:12:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Revert "MPS Upscalers Fix"

Browse Source 
This reverts commit 768b95394a.
This commit is contained in:
brkirch
2022-11-07 19:25:43 -05:00
parent 98947d173e
commit abfa22c16f
4 changed files with 4 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
modules/devices.py
View File

@@ -94,12 +94,3 @@ def autocast(disable=False):
return contextlib.nullcontext()
return torch.autocast("cuda")
# MPS workaround for https://github.com/pytorch/pytorch/issues/79383
def mps_contiguous(input_tensor, device):
return input_tensor.contiguous() if device.type == 'mps' else input_tensor
def mps_contiguous_to(input_tensor, device):
return mps_contiguous(input_tensor, device).to(device)