Commit Graph

283 Commits

Author SHA1 Message Date
Guillaume B
a1a3cfd77d Updated documentation 2020-06-22 20:41:25 +02:00
Guillaume B
12a1e0258c Addition of integration tests for Albert 2020-06-22 18:54:51 +02:00
Guillaume B
2897027ab0 Addition of AlbertForQuestionAnswering and AlbertForMultipleChoices 2020-06-22 18:29:16 +02:00
Guillaume B
2be2eecfe0 Addition of AlbertForSequenceClassification 2020-06-22 18:08:17 +02:00
Guillaume B
8b2fda9f54 Addition of AlbertForMaskedLM 2020-06-21 11:50:15 +02:00
Guillaume B
d3e91204ae Addition of AlbertMLMHead 2020-06-21 10:50:58 +02:00
Guillaume B
e1ee756581 Addition of AlbertModel 2020-06-19 17:45:04 +02:00
Guillaume B
500fbcbd56 Addition of AlbertTransformer 2020-06-19 17:30:51 +02:00
Guillaume B
9c4fdd3179 Addition of AlbertLayerGroup 2020-06-18 20:30:35 +02:00
Guillaume B
7fa19a9284 Addition of AlbertLayer 2020-06-18 20:20:48 +02:00
Guillaume B
ead102705f Addition of AlbertSelfAttention 2020-06-17 17:33:43 +02:00
Guillaume B
85e1ded9d8 Addition of Albert Embeddings (aligned with BERT and Electra) 2020-06-17 17:03:14 +02:00
Guillaume B
f4afd35ed6 albert config implementation, download scripts 2020-06-16 19:07:37 +02:00
guillaume-be
7a86436f38
Merge pull request #51 from guillaume-be/bart_xsum_addition
Addition of BART finetuned on XSum dataset
2020-06-11 10:31:32 +02:00
Guillaume B
222ef86784 Examples clean-up 2020-06-11 09:45:44 +02:00
guillaume-be
a459fa1b17
Merge pull request #49 from guillaume-be/non_mutable_generation
Updated generation pipeline to be non mutable, reworked BART caching mechanism
2020-06-06 10:54:21 +02:00
Guillaume B
bd2906fcfc Examples clean-up 2020-06-06 10:37:53 +02:00
Guillaume B
cefa0795ea BART Attention optimization 2020-06-06 09:42:01 +02:00
Guillaume B
e409bafbb7 Mutability of LayerStates 2020-06-06 08:38:15 +02:00
Guillaume B
db3d126a2b non optional LayerState fields 2020-06-05 20:49:26 +02:00
Guillaume B
e0772f42ae Updated version 2020-06-05 09:45:12 +02:00
guillaume-be
96d867b992
Merge pull request #48 from guillaume-be/sequence_classification_update
Sequence classification update
2020-06-05 09:02:59 +02:00
Guillaume B
b5c62b5d84 Updated version 2020-06-05 08:44:00 +02:00
Guillaume B
b42a14a396 Updated sentence classification to work with single input 2020-06-05 08:43:20 +02:00
Guillaume B
40a6a148a0 Migration of translation pipeline to be non mutable 2020-06-04 23:05:35 +02:00
Guillaume B
59687b5e82 Migration of BART to non mutable generator (3) 2020-06-04 22:19:42 +02:00
Guillaume B
3781040055 Migration of BART to non mutable generator (2) 2020-06-04 22:12:11 +02:00
Guillaume B
69a424d09a Migration of BART to non mutable generator (1) 2020-06-04 21:38:10 +02:00
Guillaume B
6d14e51018 Updated doctests 2020-06-04 13:12:04 +02:00
Guillaume B
69e7000aeb Definition of Cache enum and implementation for GPT2 2020-06-04 13:00:10 +02:00
Guillaume B
3c010a077b Updated generation code for GPT and GPT2 to allow non mutable models 2020-06-03 19:08:40 +02:00
guillaume-be
fa2182431c
Merge pull request #46 from guillaume-be/translation_update
Translation update
2020-05-27 16:00:14 +00:00
Guillaume B
f7c078abeb Updated documentation 2020-05-27 17:33:28 +02:00
Guillaume B
1dd522b238 Addition to English <-> Russian 2020-05-27 17:11:34 +02:00
guillaume-be
9f96c95a6a
Creation of local runtime for async resource download (#43)
* Creation of local runtime for async resource download

* CI trigger
2020-05-26 19:43:43 +02:00
Guillaume B
897833885e updated version 2020-05-25 22:34:33 +02:00
guillaume-be
d360b5cfdf
Merge pull request #44 from guillaume-be/marian_implementation
Marian implementation
2020-05-25 20:19:24 +00:00
Guillaume B
829965d68b Model resources update, documentation 2020-05-25 22:01:53 +02:00
Guillaume B
ff312a7b9e Allow for models requiring a prefix 2020-05-25 20:17:02 +02:00
Guillaume B
c315354340 Licenses update 2020-05-25 16:58:40 +02:00
Guillaume B
ff0277925e Creation of translation pipeline, added documentation, created unit tests 2020-05-25 16:20:23 +02:00
Guillaume B
67ab4fc772 Updated copyright notices 2020-05-24 21:20:21 +02:00
Guillaume B
ff2e9f2581 Update scores for padding index 2020-05-24 20:29:44 +02:00
Guillaume B
d6b5abe947 Split MarianForConditionalGeneration to own module 2020-05-24 20:15:07 +02:00
Guillaume B
cccad54194 Updates to BART for Marian compatibility 2020-05-24 17:00:04 +02:00
Guillaume B
2b498bb10a Creation of MarianGenerator 2020-05-24 15:39:17 +02:00
guillaume-be
60bffe6e8b
Merge pull request #41 from guillaume-be/multilabel_classification
Addition of multi-label classification prediction method for sequence…
2020-05-20 16:41:11 +00:00
Guillaume B
d0fc3ff40d Addition of multi-label classification prediction method for sequence classification pipeline
Version update
2020-05-20 17:51:29 +02:00
guillaume-be
749ceeecd0
Merge pull request #39 from guillaume-be/tokenizer_version_update
Support of rust_tokenizers 3.1.0
2020-05-19 16:59:39 +00:00
Guillaume B
33a623e54d Support of rust_tokenizers 3.1.0 2020-05-19 18:44:11 +02:00