Removed rust-bert
This commit is contained in:
@@ -39,7 +39,3 @@ features = ["json", "rustls-tls"]
|
||||
version = "2.1.0"
|
||||
default-features = false
|
||||
features = ["postgres", "32-column-tables", "serde_json", "r2d2", "with-deprecated"]
|
||||
|
||||
[dependencies.rust-bert]
|
||||
version = "0.21.0"
|
||||
features = ["download-libtorch"]
|
||||
@@ -311,18 +311,3 @@ fn truncate(s: &str, max_chars: usize) -> &str {
|
||||
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);
|
||||
// }
|
||||
|
||||
Reference in New Issue
Block a user