pytorch / TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
https://pytorch.org/TensorRT
BSD 3-Clause "New" or "Revised" License
2.54k stars 349 forks source link

Fixed layernorm when weight and bias is None in Stable Diffusion 3 #2936

Closed cehongwang closed 3 months ago

cehongwang commented 3 months ago

Description

Fixed layernorm when weight and bias is None in Stable Diffusion 3

Fixes # (issue)

Type of change

Checklist:

narendasan commented 3 months ago

Can you add a test case if possible?