Commit Graph

312 Commits

Author SHA1 Message Date
Guillaume B
3e0ad01a3b Updated contribution guidelines 2020-06-30 20:24:31 +02:00
Guillaume B
8eb74e5014 Addition of contribution guidelines 2020-06-30 20:20:33 +02:00
guillaume-be
7a8e1f5de3
Merge pull request #59 from guillaume-be/update_path_patch
Updated tch::nn::Path handling to be consistent across crate
2020-06-30 20:04:44 +02:00
Guillaume B
afa03cd46e Updated tch::nn::Path handling to be consistent across crate 2020-06-30 19:47:08 +02:00
guillaume-be
a067faf574
Merge pull request #58 from jerry73204/generic-path
Generic path argument
2020-06-30 19:20:06 +02:00
jerry73204
4095545f64 Use generic path 2020-06-29 16:44:49 +08:00
guillaume-be
d076ec6f77
Merge pull request #57 from guillaume-be/multiturn_conversation
Multiturn conversation
2020-06-28 11:56:15 +02:00
Guillaume B
7169e443d1 Updated doctests formatting 2020-06-28 11:19:46 +02:00
Guillaume B
23310dfc1e Added documentation, updated tests & doctests 2020-06-28 11:04:36 +02:00
Guillaume B
68def5a912 Addition of tests for conversation 2020-06-28 09:31:41 +02:00
Guillaume B
caa2ba9108 Updated conversation manager and example 2020-06-28 09:09:34 +02:00
Guillaume B
dbdc75a9bb Fixed generation bug, enable multiple batched conversations 2020-06-27 18:44:14 +02:00
Guillaume B
372c357463 Update of conversation manager and conversations 2020-06-27 18:27:43 +02:00
Guillaume B
c946791b58 Initial draft for conversation and conversation manager 2020-06-27 10:51:59 +02:00
Guillaume B
d4f9182950 Addition of history clean-up 2020-06-27 09:52:10 +02:00
Guillaume B
f9895752bd Addition of history tokens for multi-turn conversation 2020-06-27 09:32:28 +02:00
Guillaume B
28aaa94261 refactored encoding in conversation api to separate function 2020-06-26 18:19:56 +02:00
Guillaume B
c045b3212f Updated generation process to allow for direct ids input 2020-06-25 21:32:45 +02:00
Guillaume B
9081bc3318 Initial single turn conversation 2020-06-25 19:32:36 +02:00
Guillaume B
30528ca973 Download DialoGPT dependencies 2020-06-24 19:17:05 +02:00
Guillaume B
c98695c06f Updated keywords 2020-06-23 18:02:44 +02:00
guillaume-be
95d24eb232
Merge pull request #55 from guillaume-be/code_formatting
Code formatting
2020-06-23 17:51:40 +02:00
Guillaume B
cbf9c4d480 Updated doctests
Updated doctests (2)
2020-06-23 17:35:31 +02:00
Guillaume B
7fdcf6dd5d Addition of formatting checks in CI (nightly toolchain) 2020-06-23 17:06:48 +02:00
Guillaume B
4481675ec7 Addition of formatting checks in CI 2020-06-23 17:02:53 +02:00
Guillaume B
47e36c4e8c Code formatted using rustfmt 2020-06-23 16:54:46 +02:00
guillaume-be
0624a5368c
Merge pull request #54 from guillaume-be/albert_implementation
Albert implementation
2020-06-22 21:36:09 +02:00
Guillaume B
b5b6f68410 addition of albert dependencies download 2020-06-22 20:59:36 +02:00
Guillaume B
d49a1e0935 Updated documentation, addition of weights download for ALBERT 2020-06-22 20:51:01 +02:00
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