diff --git a/setup.py b/setup.py index 11b64cff2d61..103ff16e35b6 100644 --- a/setup.py +++ b/setup.py @@ -110,7 +110,7 @@ "jax>=0.4.1", "jaxlib>=0.4.1", "Jinja2", - "k-diffusion>=0.0.12", + "k-diffusion==0.0.12", "torchsde", "note_seq", "librosa", diff --git a/src/diffusers/dependency_versions_table.py b/src/diffusers/dependency_versions_table.py index f35353c49eab..ec52bcd636b9 100644 --- a/src/diffusers/dependency_versions_table.py +++ b/src/diffusers/dependency_versions_table.py @@ -17,7 +17,7 @@ "jax": "jax>=0.4.1", "jaxlib": "jaxlib>=0.4.1", "Jinja2": "Jinja2", - "k-diffusion": "k-diffusion>=0.0.12", + "k-diffusion": "k-diffusion==0.0.12", "torchsde": "torchsde", "note_seq": "note_seq", "librosa": "librosa",