released torch and trl upper bounds #17

Signed-off-by: Yu Chin Fabian Lim <[email protected]>
foundation-model-stack · Jun 30, 2024 · c681c35 · c681c35
1 parent 3392280
commit c681c35
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/plugins/framework/pyproject.toml b/plugins/framework/pyproject.toml
@@ -23,11 +23,10 @@ classifiers=[
 ]
 dependencies = [
   "numpy<2.0", # numpy needs to be bounded due to incompatiblity with current torch<2.3
-  "torch>2.2,<2.3",
+  "torch>2.2",
   "transformers<4.40",
   "peft",
   "accelerate",
-  "trl<0.9", # trl version needs to be constrained until below issue is resolved. https://github.com/foundation-model-stack/fms-hf-tuning/issues/198
 ]
 
 [tool.hatch.build.targets.wheel]