search

siliconflow / onediff

OneDiff: An out-of-the-box acceleration library for diffusion models.
https://github.com/siliconflow/onediff/wiki
Apache License 2.0
1.61k stars 99 forks source link

"LayerNormKernelImpl" not implemented for 'Half' #903

Closed liuchenbaidu closed 2 months ago

liuchenbaidu commented 4 months ago

File "/usr/local/lib/python3.9/site-packages/torch/nn/functional.py", line 2515, in layer_norm return torch.layer_norm(input, normalized_shape, weight, bias, eps, torch.backends.cudnn.enabled) RuntimeError: "LayerNormKernelImpl" not implemented for 'Half'

onediff Version: 1.1.0.dev202405220128

CuddleSabe commented 3 months ago

Maybe you should update your PyTorch edition.

strint commented 2 months ago

The problem seems in torch. Please follow @CuddleSabe 's suggestion.