[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
pull/5568/head
pre-commit-ci[bot] 8 months ago
parent ce9401ad52
commit d78817539e

@ -80,6 +80,5 @@ PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
m.def("fused_add_rms_layernorm", &fused_add_rms_layernorm, m.def("fused_add_rms_layernorm", &fused_add_rms_layernorm,
"In-place fused Add and RMS Normalization."); "In-place fused Add and RMS Normalization.");
m.def("get_cos_and_sin", &get_cos_and_sin, m.def("get_cos_and_sin", &get_cos_and_sin, "Get cos and sin from the cache.");
"Get cos and sin from the cache.");
} }

Loading…
Cancel
Save