summaryrefslogtreecommitdiff
path: root/candle-transformers/src/models/mod.rs
diff options
context:
space:
mode:
authorLaurent Mazare <laurent.mazare@gmail.com>2023-09-10 09:40:27 +0100
committerGitHub <noreply@github.com>2023-09-10 09:40:27 +0100
commitd3f05eae8c4f2df186b46e433be101ac39fceca5 (patch)
tree6ffc43595caec3007fe28efd3bafc7acbdde6e94 /candle-transformers/src/models/mod.rs
parent258ac32c3868d4103e90df19af99a3e13c805c4e (diff)
downloadcandle-d3f05eae8c4f2df186b46e433be101ac39fceca5.tar.gz
candle-d3f05eae8c4f2df186b46e433be101ac39fceca5.tar.bz2
candle-d3f05eae8c4f2df186b46e433be101ac39fceca5.zip
Move some models to candle-transformers so that it's easier to re-use. (#794)
* Move some models to candle-transformers so that they can be shared. * Also move falcon. * Move Llama. * Move whisper (partial).
Diffstat (limited to 'candle-transformers/src/models/mod.rs')
-rw-r--r--candle-transformers/src/models/mod.rs6
1 files changed, 5 insertions, 1 deletions
diff --git a/candle-transformers/src/models/mod.rs b/candle-transformers/src/models/mod.rs
index 8b137891..1b3dcf25 100644
--- a/candle-transformers/src/models/mod.rs
+++ b/candle-transformers/src/models/mod.rs
@@ -1 +1,5 @@
-
+pub mod bert;
+pub mod bigcode;
+pub mod falcon;
+pub mod llama;
+pub mod whisper;