From 48f4340b149f5438c4c21df18bae3e666b174b6b Mon Sep 17 00:00:00 2001 From: Kangyan-Zhou Date: Sun, 25 Jan 2026 23:32:39 -0800 Subject: [PATCH] Exclude some diffusion package for ARM in docker release (#17745) --- python/pyproject.toml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/pyproject.toml b/python/pyproject.toml index 5161c98db..86d6d7230 100755 --- a/python/pyproject.toml +++ b/python/pyproject.toml @@ -93,8 +93,8 @@ diffusion = [ "moviepy>=2.0.0", "opencv-python-headless==4.10.0.84", "remote-pdb", - "st_attn==0.0.7", - "vsa==0.0.4", + "st_attn==0.0.7 ; platform_machine != 'aarch64' and platform_machine != 'arm64'", + "vsa==0.0.4 ; platform_machine != 'aarch64' and platform_machine != 'arm64'", "runai_model_streamer", "cache-dit==1.2.0" ]