fix: Remove unused PyTorch dependency

Fixes instructlab#478 Signed-off-by: Fabien Dupont <[email protected]>
aakankshaduggal · Jan 15, 2025 · 0d11884 · 0d11884
1 parent 725e5ab
commit 0d11884
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -22,6 +22,8 @@ Any users that were specifying custom pipeline configs (instead of using the def
 
 ### Fixes
 
+* The PyTorch dependency is removed, because SDG doesn't directly use PyTorch. The test suite still depends on `instructlab` core, which depends on PyTorch.
+
 ## v0.6.3
 
 ### Fixes

diff --git a/requirements.txt b/requirements.txt
@@ -17,6 +17,5 @@ openai>=1.13.3,<2.0.0
 sentencepiece>=0.2.0
 tabulate>=0.9.0
 tenacity>=8.3.0,!=8.4.0
-torch>=2.3.0,<2.6.0
 transformers>=4.41.2
 xdg-base-dirs>=6.0.1