updating docstring for AutoModel

pytorch_transformers/modeling_utils.py

@@ -59,6 +59,12 @@ if not six.PY2:
             fn.__doc__ = ''.join(docstr) + fn.__doc__
             return fn
         return docstring_decorator
+
+    def add_end_docstrings(*docstr):
+        def docstring_decorator(fn):
+            fn.__doc__ = fn.__doc__ + ''.join(docstr)
+            return fn
+        return docstring_decorator
 else:
     # Not possible to update class docstrings on python2
     def add_start_docstrings(*docstr):
@@ -66,6 +72,11 @@ else:
             return fn
         return docstring_decorator
 
+    def add_end_docstrings(*docstr):
+        def docstring_decorator(fn):
+            return fn
+        return docstring_decorator
+
 
 class PretrainedConfig(object):
     r""" Base class for all configuration classes.