Skip to content

Commit 811b35e

Browse files
committed
Added other missing resources
1 parent 7e00a22 commit 811b35e

File tree

1 file changed

+17
-2
lines changed

1 file changed

+17
-2
lines changed

src/t5/t5_model.rs

+17-2
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,11 @@ impl T5ModelResources {
6666
"godel-v1-1-base/model",
6767
"https://huggingface.co/microsoft/GODEL-v1_1-base-seq2seq/resolve/main/rust_model.ot",
6868
);
69+
/// Shared under MIT license by the Microsoft team at <https://huggingface.co/microsoft/GODEL-v1_1-base-seq2seq>. Modified with conversion to C-array format.
70+
pub const GODEL_V1_1_LARGE: (&'static str, &'static str) = (
71+
"godel-v1-1-base/model",
72+
"https://huggingface.co/microsoft/GODEL-v1_1-large-seq2seq/resolve/main/rust_model.ot",
73+
);
6974
}
7075

7176
impl T5ConfigResources {
@@ -89,6 +94,11 @@ impl T5ConfigResources {
8994
"godel-v1-1-base/config",
9095
"https://huggingface.co/microsoft/GODEL-v1_1-base-seq2seq/resolve/main/config.json",
9196
);
97+
/// Shared under MIT license by the Microsoft team at <https://huggingface.co/microsoft/GODEL-v1_1-base-seq2seq>. Modified with conversion to C-array format.
98+
pub const GODEL_V1_1_LARGE: (&'static str, &'static str) = (
99+
"godel-v1-1-base/config",
100+
"https://huggingface.co/microsoft/GODEL-v1_1-large-seq2seq/resolve/main/config.json",
101+
);
92102
}
93103

94104
impl T5VocabResources {
@@ -107,10 +117,15 @@ impl T5VocabResources {
107117
"sentence-t5-base/spiece",
108118
"https://huggingface.co/sentence-transformers/sentence-t5-base/resolve/main/spiece.model",
109119
);
110-
/// Shared under MIT license by the Microsoft team at <https://huggingface.co/microsoft/GODEL-v1_1-base-seq2seq>. Modified with conversion to C-array format.
120+
/// Shared under Apache 2.0 license by the Google team at <https://github.com/google-research/text-to-text-transfer-transformer>.
111121
pub const GODEL_V1_1_BASE: (&'static str, &'static str) = (
112122
"godel-v1-1-base/spiece",
113-
"https://huggingface.co/microsoft/GODEL-v1_1-base-seq2seq/resolve/main/spiece.model",
123+
"https://huggingface.co/t5-base/resolve/main/spiece.model",
124+
);
125+
/// Shared under Apache 2.0 license by the Google team at <https://github.com/google-research/text-to-text-transfer-transformer>.
126+
pub const GODEL_V1_1_LARGE: (&'static str, &'static str) = (
127+
"godel-v1-1-base/spiece",
128+
"https://huggingface.co/t5-large/resolve/main/spiece.model",
114129
);
115130
}
116131

0 commit comments

Comments
 (0)