Pin torch to CUDA 12 compatible version

2026-04-02 18:11:57 +02:00
parent 4f3c568123
commit 7ad1a08692
2 changed files with 89 additions and 215 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -8,7 +8,7 @@ dependencies = [
  "boto3",
  "python-dotenv",
  "elevenlabs",
-  "torch",
+  "torch==2.5.1",
  "transformers",
  "diffusers",
  "accelerate",