diff options
author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-09-10 09:40:27 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-09-10 09:40:27 +0100 |
commit | d3f05eae8c4f2df186b46e433be101ac39fceca5 (patch) | |
tree | 6ffc43595caec3007fe28efd3bafc7acbdde6e94 /candle-transformers/src/models/mod.rs | |
parent | 258ac32c3868d4103e90df19af99a3e13c805c4e (diff) | |
download | candle-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.rs | 6 |
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; |