fix custom cache dir (#2226)
* fix to not ignore HUGGINGFACE_HUB_CACHE in cache * delete printlns * delete newlines * maybe fix trailing whitespace
This commit is contained in:
parent
5a65066922
commit
457fb0a188
|
@ -210,7 +210,11 @@ async fn main() -> Result<(), RouterError> {
|
||||||
}
|
}
|
||||||
let api = if use_api {
|
let api = if use_api {
|
||||||
if std::env::var("HF_HUB_OFFLINE") == Ok("1".to_string()) {
|
if std::env::var("HF_HUB_OFFLINE") == Ok("1".to_string()) {
|
||||||
let cache = Cache::default();
|
let cache = std::env::var("HUGGINGFACE_HUB_CACHE")
|
||||||
|
.map_err(|_| ())
|
||||||
|
.map(|cache_dir| Cache::new(cache_dir.into()))
|
||||||
|
.unwrap_or_else(|_| Cache::default());
|
||||||
|
|
||||||
tracing::warn!("Offline mode active using cache defaults");
|
tracing::warn!("Offline mode active using cache defaults");
|
||||||
Type::Cache(cache)
|
Type::Cache(cache)
|
||||||
} else {
|
} else {
|
||||||
|
|
Loading…
Reference in New Issue