From b4e6f8acad64b1a8567a2e63991b019b8ef0a6d4 Mon Sep 17 00:00:00 2001 From: Alex Cheema Date: Thu, 13 Feb 2025 22:55:09 +0000 Subject: [PATCH] always log download errors. some people eg cant access huggingface which causes confusion --- exo/download/new_shard_download.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/exo/download/new_shard_download.py b/exo/download/new_shard_download.py index 21b3f046..5e6531f0 100644 --- a/exo/download/new_shard_download.py +++ b/exo/download/new_shard_download.py @@ -134,6 +134,8 @@ async def download_file_with_retry(repo_id: str, revision: str, path: str, targe try: return await _download_file(repo_id, revision, path, target_dir, on_progress) except Exception as e: if isinstance(e, FileNotFoundError) or attempt == n_attempts - 1: raise e + print(f"Download error on attempt {attempt}/{n_attempts} for {repo_id=} {revision=} {path=} {target_dir=}") + traceback.print_exc() await asyncio.sleep(min(8, 0.1 * (2 ** attempt))) async def _download_file(repo_id: str, revision: str, path: str, target_dir: Path, on_progress: Callable[[int, int], None] = lambda _, __: None) -> Path: