From e1e2fd73f3429007b29b5b3f35341b645e5007e3 Mon Sep 17 00:00:00 2001 From: Radu Swigler Date: Thu, 3 Sep 2026 03:55:23 -0400 Subject: [PATCH] fix: remove torchvision from SIZE_PROHIBITIVE_PACKAGES (#353) torchvision is not present in GPU base images (only torch is), so excluding it causes silent ImportError at runtime when users explicitly request it via dependencies=["torchvision"]. The package is also only 7.3 MB now, not the 50 MB cited in the comment. Closes #353 --- src/runpod_flash/cli/commands/build.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/runpod_flash/cli/commands/build.py b/src/runpod_flash/cli/commands/build.py index 4f00b9a9..0d408472 100644 --- a/src/runpod_flash/cli/commands/build.py +++ b/src/runpod_flash/cli/commands/build.py @@ -168,7 +168,6 @@ def compute_source_fingerprint(project_dir: Path, files: list[Path]) -> str: SIZE_PROHIBITIVE_PACKAGES: frozenset[str] = frozenset( { "torch", # ~500 MB - "torchvision", # ~50 MB, requires torch "torchaudio", # ~30 MB, requires torch "triton", # ~150 MB, CUDA compiler }