From 5ff802512c4d5e20b68852376965922cfc124194 Mon Sep 17 00:00:00 2001 From: "promptless[bot]" <179508745+promptless[bot]@users.noreply.github.com> Date: Mon, 23 Mar 2026 17:12:29 +0000 Subject: [PATCH] Remove numpy from auto-excluded build packages --- flash/cli/build.mdx | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/flash/cli/build.mdx b/flash/cli/build.mdx index 88396834..c194741e 100644 --- a/flash/cli/build.mdx +++ b/flash/cli/build.mdx @@ -83,8 +83,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` These packages are excluded at archive time, so you don't need to specify them manually. @@ -100,8 +99,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` | + @@ -125,7 +125,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