From 93de0fbdacd906faad7716ded83c450a95552393 Mon Sep 17 00:00:00 2001 From: Gabriel Date: Thu, 25 Jun 2026 11:43:41 -0500 Subject: [PATCH] Constrain numba test dependency --- setup.py | 2 ++ src/diffusers/dependency_versions_table.py | 1 + 2 files changed, 3 insertions(+) diff --git a/setup.py b/setup.py index a9bafaff5399..0dc00862db67 100644 --- a/setup.py +++ b/setup.py @@ -114,6 +114,7 @@ "torchsde", "note_seq", "librosa", + "numba>=0.59.0", "numpy", "parameterized", "peft>=0.17.0", @@ -228,6 +229,7 @@ def run(self): "Jinja2", "invisible-watermark", "librosa", + "numba", "parameterized", "protobuf", "pytest", diff --git a/src/diffusers/dependency_versions_table.py b/src/diffusers/dependency_versions_table.py index 3aac2f280af6..95dd85e1f1ac 100644 --- a/src/diffusers/dependency_versions_table.py +++ b/src/diffusers/dependency_versions_table.py @@ -21,6 +21,7 @@ "torchsde": "torchsde", "note_seq": "note_seq", "librosa": "librosa", + "numba": "numba>=0.59.0", "numpy": "numpy", "parameterized": "parameterized", "peft": "peft>=0.17.0",