Skip to content

docs: Remove numpy from auto-excluded build packages #581

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
muhsinking merged 2 commits into from
Mar 23, 2026
Merged

docs: Remove numpy from auto-excluded build packages #581

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
5ff8025
Remove numpy from auto-excluded build packages
promptless[bot] Mar 23, 2026
06994f6
Merge branch 'main' into promptless/flash-numpy-build-exclusion
muhsinking Mar 23, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 5 additions & 5 deletions flash/cli/build.mdx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -84,8 +84,7 @@ Image tags follow the pattern `py{version}-{tag}` (for example, `runpod/flash:py

Runpod Serverless has a **1.5GB deployment limit**. Flash automatically excludes packages that are pre-installed in the base image:

- `torch`, `torchvision`, `torchaudio`
- `numpy`, `triton`
- `torch`, `torchvision`, `torchaudio`, `triton`
Copy link
Contributor Author

@promptless promptless bot Mar 23, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Citation: Updated to match the new SIZE_PROHIBITIVE_PACKAGES constant in src/runpod_flash/cli/commands/build.py. Each package is annotated with its approximate size: torch (~500 MB), torchvision (~50 MB), torchaudio (~30 MB), triton (~150 MB).
View source


These packages are excluded at archive time, so you don't need to specify them manually.

Expand All @@ -101,8 +100,9 @@ flash build --exclude scipy,pandas

| Resource type | Base image | Auto-excluded packages |
|--------------|------------|------------------------|
| GPU | PyTorch base | `torch`, `torchvision`, `torchaudio`, `numpy`, `triton` |
| CPU | Python slim | `torch`, `torchvision`, `torchaudio`, `numpy`, `triton` |
| GPU | PyTorch base | `torch`, `torchvision`, `torchaudio`, `triton` |
| CPU | Python slim | `torch`, `torchvision`, `torchaudio`, `triton` |


<Tip>

Copy link
Contributor Author

@promptless promptless bot Mar 23, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Citation: PR #263 renames BASE_IMAGE_PACKAGES to SIZE_PROHIBITIVE_PACKAGES and removes numpy from the list. The root cause was that the blacklist was defined by GPU base image contents, not size constraints, which silently broke CPU endpoints using python-slim that needed numpy. See changes in src/runpod_flash/cli/commands/build.py.
View source

Expand All @@ -126,7 +126,7 @@ def my_function(data):

### Archive is too large

Base image packages (`torch`, `numpy`, `triton`, etc.) are auto-excluded. If the archive is still too large, use `--exclude` to skip additional packages or `--no-deps` to skip transitive dependencies:
Some CUDA packages (`torch`, `torchvision`, `torchaudio`, `triton`) are auto-excluded. If the archive is still too large, use `--exclude` to skip additional packages or `--no-deps` to skip transitive dependencies:

```bash
flash build --exclude scipy,pandas
Expand Down
Loading