Skip to content

Commit ebb4e35

Browse files
authored
change dependency in gemma accuracy (#361)
1 parent d0cf2a5 commit ebb4e35

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

example/gemma/gemma.patch

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -103,3 +103,16 @@ index 00000000..732185c4
103103
+ def _model_generate(self, context, max_length, eos_token_id):
104104
+ # Isn't used because we override greedy_until
105105
+ raise NotImplementedError()
106+
diff --git a/setup.py b/setup.py
107+
index 5db43c17..ea8b627a 100644
108+
--- a/setup.py
109+
+++ b/setup.py
110+
@@ -21,7 +21,7 @@ setuptools.setup(
111+
],
112+
python_requires=">=3.9",
113+
install_requires=[
114+
- "datasets>=2.0.0",
115+
+ "datasets>=2.20.0",
116+
"jsonlines",
117+
"numexpr",
118+
"openai>=0.6.4",

0 commit comments

Comments
 (0)