added version constrained for torch and torchaudio to be compatible with flash-attn

c3cface8 · Deep-unlearning · 446f8ae4 · c3cface8
Commit c3cface8 authored May 16, 2025 by Deep-unlearning
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

pyproject.toml pyproject.toml +2 -2

No files found.
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -12,8 +12,8 @@ license     = {text = "MIT"}
 authors     = [{name = "MoonshotAI", email = "contact@moonshot.ai"}]
 dependencies = [
-  "torch",
+  "torch<=2.6.0",
-  "torchaudio",
+  "torchaudio<=2.6.0",
  "flash-attn",
  "soundfile",
  "librosa",