Skip to content

Commit a3f484e

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

File tree

2 files changed

+19
-2
lines changed

2 files changed

+19
-2
lines changed

src/pipelines/conversation.rs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -925,6 +925,8 @@ impl ConversationModel {
925925

926926
let mut output = HashMap::with_capacity(active_uuid.len());
927927

928+
println!("generated: {:#?}, prompt_ids: {:#?}", &generated, &prompt_ids);
929+
928930
for (
929931
((conversation, (generated_sequence, conversation_promp_ids)), uuid),
930932
removed_padding,

src/t5/t5_model.rs

Lines changed: 17 additions & 2 deletions
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-large/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-large/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-large/spiece",
128+
"https://huggingface.co/t5-large/resolve/main/spiece.model",
114129
);
115130
}
116131

0 commit comments

Comments
 (0)