Commit Graph

64 Commits

Author SHA1 Message Date
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
47e36c4e8c Code formatted using rustfmt 2020-06-23 16:54:46 +02:00
Guillaume B
12a1e0258c Addition of integration tests for Albert 2020-06-22 18:54:51 +02:00
Guillaume B
e0772f42ae Updated version 2020-06-05 09:45:12 +02:00
Guillaume B
3781040055 Migration of BART to non mutable generator (2) 2020-06-04 22:12:11 +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 B
829965d68b Model resources update, documentation 2020-05-25 22:01:53 +02:00
Guillaume B
ff0277925e Creation of translation pipeline, added documentation, created unit tests 2020-05-25 16:20:23 +02:00
Guillaume B
bb94510075 Updated integration tests and formatting 2020-05-09 22:21:58 +02:00
Guillaume B
c6f5cdd859 Updated documentation 2020-05-03 13:37:18 +02:00
Guillaume B
9f608f4374 Addition of tests for Electra 2020-05-03 11:47:16 +02:00
Guillaume B
029d4bd47c Addition of Electra resources 2020-05-03 10:05:32 +02:00
Guillaume B
194a447b06 Updated documentation 2020-04-26 11:49:40 +02:00
Guillaume B
cb254995fd Addition of dependencies download example 2020-04-26 10:51:29 +02:00
Guillaume B
0a85d5ba21 Updated NER Pipeline 2020-04-26 09:13:13 +02:00
Guillaume B
2aeace4a6e Updated Question Answering Pipeline 2020-04-26 09:08:04 +02:00
Guillaume B
69b819ce5e Updated Summarization and Sentiment pipelines 2020-04-26 08:57:20 +02:00
Guillaume B
43091c8e24 Updated BART Base model var store path 2020-04-25 20:35:37 +02:00
Guillaume B
f7aa61f131 Updated Roberta models with new configs and dependency loaders 2020-04-25 17:53:04 +02:00
Guillaume B
ce4041e1d2 Updated GPT models with new configs and dependency loaders 2020-04-25 17:43:44 +02:00
Guillaume B
b5b772c488 Updated GPT2 models with new configs and dependency loaders 2020-04-25 17:34:59 +02:00
Guillaume B
9168e898a9 Updated BART models with new configs and dependency loaders 2020-04-25 17:27:33 +02:00
Guillaume B
5f4a8f8bdf Updated BERT models with new configs and dependency loaders 2020-04-25 17:01:46 +02:00
Guillaume B
3d5a3669ed Updated DistilBert models with new configs and dependency loaders 2020-04-25 16:42:56 +02:00
Guillaume B
cdfe14a3a4 Update of dependencies 2020-04-25 15:41:53 +02:00
Guillaume B
eaccc23be4 Download utilities 2020-04-25 11:31:02 +02:00
Guillaume B
c65943ab2f Updated model weight download 2020-04-24 17:04:28 +02:00
guillaume-be
03f642fc68
Shared embeddings for BART (#18) 2020-04-05 15:29:19 +02:00
Guillaume B
760676e5f0 skipping expensive bart tests for CI 2020-04-05 13:49:39 +02:00
Guillaume B
dc727e0baa Updated documentation 2020-04-05 11:40:08 +02:00
Guillaume B
6c7353de6e Addition of summarization pipeline and integration tests 2020-04-05 10:21:45 +02:00
Guillaume B
dbf6841610 Optimization of generation pipeline 2020-04-04 09:40:13 +02:00
Guillaume B
6aa6a4b2a2 Encoder input for summarization validated 2020-04-03 17:11:09 +02:00
Guillaume B
b8fe6a41e5 Updated language generation API to work with mutable models 2020-04-03 14:48:08 +02:00
Guillaume B
eea1667f26 Added documentation for pipelines 2020-03-27 16:17:17 +01:00
Guillaume B
6c6e5526ec Completed documentation for BERT 2020-03-22 15:39:22 +01:00
Guillaume B
54b10c54a7 Updated lib structure and visibility 2020-03-22 09:54:49 +01:00
Guillaume B
f17b0d7da8 Lib clean-up and doc landing page 2020-03-21 16:53:15 +01:00
Guillaume B
24335a5883 Lib clean-up (default binary deleted, re-exports removed) 2020-03-21 11:35:41 +01:00
Guillaume B
344e27c927 Performance optimization for language generation 2020-03-17 20:06:45 +01:00
Guillaume B
64a01e3f73 End-to-end generation completed, updated README, bumped version 2020-03-16 16:17:23 +01:00
Guillaume B
2621e0dc5d Added tests for GPT2 and OpenAI GPT generation 2020-03-15 21:09:46 +01:00
Guillaume B
2d63fc3d18 Added tests for GPT2 2020-03-08 17:55:47 +01:00
Guillaume B
1923af0ac7 initial structure for generation 2020-03-08 17:08:18 +01:00
Guillaume B
996fdb159f Updated QA for improved batching 2020-03-07 15:21:02 +01:00
Guillaume B
999f6e52aa initial commit for batch processing of QA inputs 2020-03-06 21:40:32 +01:00
Guillaume B
89b486fb12 Added integration test for OpenAI GPT 2020-03-01 11:29:57 +01:00