diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2024-04-01 20:58:34 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-04-01 20:58:34 +0200 |
| commit | be9c200cbb16b59fe1f1e8c0f606981412c9b757 (patch) | |
| tree | 4a2cdbb5d388f4cfaa42b53a9297575174baa2d3 /candle-examples/examples/t5/main.rs | |
| parent | ea0d8d3753b53a936c472c30ae5dc0d52bfa81fa (diff) | |
| download | candle-be9c200cbb16b59fe1f1e8c0f606981412c9b757.tar.gz candle-be9c200cbb16b59fe1f1e8c0f606981412c9b757.tar.bz2 candle-be9c200cbb16b59fe1f1e8c0f606981412c9b757.zip | |
Expose the t5 config fields + allow t5-large. (#1987)
Diffstat (limited to 'candle-examples/examples/t5/main.rs')
| -rw-r--r-- | candle-examples/examples/t5/main.rs | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/candle-examples/examples/t5/main.rs b/candle-examples/examples/t5/main.rs index 34ae0ead..902282c1 100644 --- a/candle-examples/examples/t5/main.rs +++ b/candle-examples/examples/t5/main.rs @@ -22,6 +22,7 @@ const DTYPE: DType = DType::F32; enum Which { T5Base, T5Small, + T5Large, T5_3B, Mt5Base, Mt5Small, @@ -108,6 +109,7 @@ impl T5ModelBuilder { let (default_model, default_revision) = match args.which { Which::T5Base => ("t5-base", "main"), Which::T5Small => ("t5-small", "refs/pr/15"), + Which::T5Large => ("t5-large", "main"), Which::T5_3B => ("t5-3b", "main"), Which::Mt5Base => ("google/mt5-base", "refs/pr/5"), Which::Mt5Small => ("google/mt5-small", "refs/pr/6"), |