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

This reverts commit c3cface8.

Revert "added version constrained for torch and torchaudio to be compatible with flash-attn"
This reverts commit c3cface8.
9a54b8e4 · Deep-unlearning · c3cface8 · 9a54b8e4
Commit 9a54b8e4 authored May 16, 2025 by Deep-unlearning
Show 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<=2.6.0",
-  "torchaudio<=2.6.0",
+  "torch",
+  "torchaudio",
  "flash-attn",
  "soundfile",
  "librosa",