init active_downloadsa

29f154597c5524231b93cac3a77c8209cefdb8e8 · 2024-08-07 13:59:19 +0100 · Alex Cheema

Files touched

M exo/download/hf/hf_shard_download.py

Diff

commit 29f154597c5524231b93cac3a77c8209cefdb8e8
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Wed Aug 7 13:59:19 2024 +0100

    init active_downloadsa
---
 exo/download/hf/hf_shard_download.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/exo/download/hf/hf_shard_download.py b/exo/download/hf/hf_shard_download.py
index b782e1a0..820ca065 100644
--- a/exo/download/hf/hf_shard_download.py
+++ b/exo/download/hf/hf_shard_download.py
@@ -10,7 +10,7 @@ from exo.helpers import AsyncCallbackSystem, DEBUG
 
 class HFShardDownloader(ShardDownloader):
     def __init__(self):
-        self.active_downloads: Dict[Shard, asyncio.Task] = []
+        self.active_downloads: Dict[Shard, asyncio.Task] = {}
         self._on_progress = AsyncCallbackSystem[str, Tuple[Shard, RepoProgressEvent]]()
 
     async def ensure_shard(self, shard: Shard) -> Path:

← 6bddb2a9 download edge cases · back to Exo · fix shard download 7ec660bb →