aboutsummaryrefslogtreecommitdiff
path: root/javascript/profilerVisualization.js
diff options
context:
space:
mode:
authorhidenorly <twitte.harold@gmail.com>2023-11-29 04:45:04 +0900
committerhidenorly <twitte.harold@gmail.com>2023-11-29 04:45:04 +0900
commita0096c58977c01ddc6a2b83a8a7b64da6fd4a51e (patch)
tree672db877f15361f144fc4e27531132917d8070f1 /javascript/profilerVisualization.js
parent39eae9f009c8302eed77b0942e1e634f6125d53e (diff)
Add FP32 fallback support on torch.nn.functional.interpolate
This tries to execute interpolate with FP32 if it failed. Background is that on some environment such as Mx chip MacOS devices, we get error as follows: ``` "torch/nn/functional.py", line 3931, in interpolate return torch._C._nn.upsample_nearest2d(input, output_size, scale_factors) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ RuntimeError: "upsample_nearest2d_channels_last" not implemented for 'Half' ``` In this case, ```--no-half``` doesn't help to solve. Therefore this commits add the FP32 fallback execution to solve it. Note that the ```upsample_nearest2d``` is called from ```torch.nn.functional.interpolate```. And the fallback for torch.nn.functional.interpolate is necessary at ```modules/sd_vae_approx.py``` 's ```VAEApprox.forward``` ```repositories/stable-diffusion-stability-ai/ldm/modules/diffusionmodules/openaimodel.py``` 's ```Upsample.forward```
Diffstat (limited to 'javascript/profilerVisualization.js')
0 files changed, 0 insertions, 0 deletions