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"
 ]