add import math (#9346)

huggingface · Dec 29, 2020 · 912f688 · 912f688
1 parent 785e52c
commit 912f688
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/...r-template-{{cookiecutter.modelname}}/modeling_tf_{{cookiecutter.lowercase_modelname}}.py b/...r-template-{{cookiecutter.modelname}}/modeling_tf_{{cookiecutter.lowercase_modelname}}.py
@@ -1400,6 +1400,7 @@ def call(
         )
 
 {% else %}
+import math
 import random
 from typing import Dict, Optional, Tuple, Union
 

diff --git a/...tter-template-{{cookiecutter.modelname}}/modeling_{{cookiecutter.lowercase_modelname}}.py b/...tter-template-{{cookiecutter.modelname}}/modeling_{{cookiecutter.lowercase_modelname}}.py
@@ -1513,6 +1513,7 @@ def forward(
             attentions=outputs.attentions,
         )
 {% else %}
+import math
 import random
 from typing import Optional, Tuple