From 514b171f75f73bf81d9a67f11e39437a53e67687 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Thu, 6 Jul 2023 14:21:35 +0200
Subject: [PATCH] Getting tokio tasks stuck on smaller machines.

---
 candle-hub/Cargo.toml       | 2 --
 candle-hub/src/api/tokio.rs | 2 +-
 2 files changed, 1 insertion(+), 3 deletions(-)

diff --git a/candle-hub/Cargo.toml b/candle-hub/Cargo.toml
index 18e5d6c2..3da7d6ec 100644
--- a/candle-hub/Cargo.toml
+++ b/candle-hub/Cargo.toml
@@ -27,5 +27,3 @@ tokio-test = "0.4.2"
 default = ["online"]
 online = ["reqwest/blocking", "dep:serde", "dep:serde_json", "dep:indicatif", "dep:num_cpus"]
 tokio = ["online", "dep:tokio", "dep:futures"]
-
-
diff --git a/candle-hub/src/api/tokio.rs b/candle-hub/src/api/tokio.rs
index bf0c2c41..dc8f682e 100644
--- a/candle-hub/src/api/tokio.rs
+++ b/candle-hub/src/api/tokio.rs
@@ -131,7 +131,7 @@ impl ApiBuilder {
             url_template: "{endpoint}/{repo_id}/resolve/{revision}/{filename}".to_string(),
             cache,
             token,
-            max_files: 100,
+            max_files: num_cpus::get(),
             chunk_size: 10_000_000,
             parallel_failures: 0,
             max_retries: 0,