bsherriff/siren
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Removed rust-bert

Browse Source
This commit is contained in:
Benjamin Sherriff
2023-07-07 18:23:30 -04:00
parent 086a059fdb
commit 8fae4022ab
2 changed files with 1 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
Cargo.toml
View File

@@ -38,8 +38,4 @@ features = ["json", "rustls-tls"]
[dependencies.diesel] [dependencies.diesel]
version = "2.1.0" version = "2.1.0"
default-features = false default-features = false
features = ["postgres", "32-column-tables", "serde_json", "r2d2", "with-deprecated"] features = ["postgres", "32-column-tables", "serde_json", "r2d2", "with-deprecated"]
[dependencies.rust-bert]
version = "0.21.0"
features = ["download-libtorch"]

15
src/commands/oai.rs
View File

@@ -311,18 +311,3 @@ fn truncate(s: &str, max_chars: usize) -> &str {
Some((idx, _)) => &s[..idx], Some((idx, _)) => &s[..idx],
} }
} }
// fn summarization_model() -> Result<SummarizationModel, RustBertError> {
// let config_resource = RemoteResource::from_pretrained(T5ConfigResources::T5_SMALL);
// let vocab_resource = RemoteResource::from_pretrained(T5VocabResources::T5_SMALL);
// let weights_resource = RemoteResource::from_pretrained(T5ModelResources::T5_SMALL);
// let summarization_config = SummarizationConfig::new(
// ModelType::T5,
// ModelResource::Torch(Box::new(weights_resource)),
// config_resource,
// vocab_resource,
// None
// );
// return SummarizationModel::new(summarization_config);
// }