Tanh torch warnings

14cdeee7 · Aryansh Omray · Julien Chaumond · 16469fed · 14cdeee7
Commit 14cdeee7 authored Apr 16, 2020 by Aryansh Omray Committed by Julien Chaumond Apr 16, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/activations.py src/transformers/activations.py +1 -1

No files found.
--- a/src/transformers/activations.py
+++ b/src/transformers/activations.py
@@ -47,7 +47,7 @@ ACT2FN = {
    "relu": F.relu,
    "swish": swish,
    "gelu": gelu,
-    "tanh": F.tanh,
+    "tanh": torch.tanh,
    "gelu_new": gelu_new,
 }