Merge pull request #78 from mozilla-ai/pin_ray

Bumping the library version and pinning lm-eval-harness to a PyPI Distribution
mozilla-ai · Mar 11, 2024 · a19ea24 · a19ea24
2 parents 045ca11 + 6bd85b9
commit a19ea24
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "lm-buddy"
-version = "0.2.7"
+version = "0.2.8"
 authors = [
     { name = "Sean Friedowitz", email = "[email protected]" },
     { name = "Aaron Gonzales", email = "[email protected]" },
@@ -35,7 +35,7 @@ dependencies = [
     "trl==0.7.10",
     "bitsandbytes==0.42.0",
     # Evaluation frameworks
-    "lm-eval @ git+https://github.com/EleutherAI/lm-evaluation-harness@45823914a3e445076a1ba12c9b1501c98e3dd20d",
+    "lm-eval==0.4.1",
     "einops==0.7.0",
 ]