未验证 提交 23e5133c 编写于 作者: M Michael Wyatt 提交者: GitHub

update for lm-eval==0.3.0 (#2713)

Co-authored-by: NJeff Rasley <jerasley@microsoft.com>
上级 0b549ad7
...@@ -38,12 +38,6 @@ jobs: ...@@ -38,12 +38,6 @@ jobs:
pip install .[dev,1bit,autotuning,inf] pip install .[dev,1bit,autotuning,inf]
ds_report ds_report
- name: Install lm-eval
run: |
pip install git+https://github.com/EleutherAI/lm-evaluation-harness
# This is required until lm-eval makes a new release. v0.2.0 is
# broken for latest version of transformers
- name: Python environment - name: Python environment
run: | run: |
pip list pip list
......
google google
lm-eval>=0.2.0 lm-eval==0.3.0
protobuf protobuf
transformers transformers
transformers[sentencepiece] transformers[sentencepiece]
...@@ -427,7 +427,7 @@ class TestInjectionPolicy(DistributedTest): ...@@ -427,7 +427,7 @@ class TestInjectionPolicy(DistributedTest):
"gpt2-xl"], "gpt2-xl"],
), ),
) )
@pytest.mark.parametrize("task", ["lambada"]) @pytest.mark.parametrize("task", ["lambada_standard"])
class TestLMCorrectness(DistributedTest): class TestLMCorrectness(DistributedTest):
world_size = 1 world_size = 1
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册