diff --git a/src/exo/download/coordinator.py b/src/exo/download/coordinator.py index 5b6026be..0a759b3b 100644 --- a/src/exo/download/coordinator.py +++ b/src/exo/download/coordinator.py @@ -388,7 +388,9 @@ class DownloadCoordinator: if model_id in self.active_downloads: continue - if progress.status == "complete": + if progress.status == "complete" or ( + progress.downloaded.in_bytes >= progress.total.in_bytes > 0 + ): status: DownloadProgress = DownloadCompleted( node_id=self.node_id, shard_metadata=progress.shard,