SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

add base model classes to bart subclassed models (#9230)

Browse Source 
* add base model classes to  bart subclassed models

* add doc
This commit is contained in:
Suraj Patil
2020-12-21 19:56:46 +05:30
committed by GitHub
parent 08abdabda1
commit f4432b7e01
15 changed files with 134 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
docs/source/model_doc/blenderbot.rst
View File

@@ -100,6 +100,15 @@ BlenderbotSmallTokenizer
:members:
BlenderbotModel
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
See :obj:`transformers.BartModel` for arguments to `forward` and `generate`
.. autoclass:: transformers.BlenderbotModel
:members:
BlenderbotForConditionalGeneration
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~