"vscode:/vscode.git/clone" did not exist on "056bea45a171df682a75943d6733a3c6a9d1b173"
Commit 84f1a131 authored by Alexander's avatar Alexander
Browse files

Merged with master

parents eab96f01 c1bd72c7
...@@ -21,7 +21,7 @@ This project provides a unified framework to test generative language models on ...@@ -21,7 +21,7 @@ This project provides a unified framework to test generative language models on
Features: Features:
- 200+ tasks implemented. See the [task-table](./docs/task_table.md) for a complete list. - 200+ subtasks / evaluation settings implemented. See the [task-table](./docs/task_table.md) for a complete list.
- Support for models loaded via [transformers](https://github.com/huggingface/transformers/) (including quantization via [AutoGPTQ](https://github.com/PanQiWei/AutoGPTQ)), [GPT-NeoX](https://github.com/EleutherAI/gpt-neox), and [Megatron-DeepSpeed](https://github.com/microsoft/Megatron-DeepSpeed/), with a flexible tokenization-agnostic interface. - Support for models loaded via [transformers](https://github.com/huggingface/transformers/) (including quantization via [AutoGPTQ](https://github.com/PanQiWei/AutoGPTQ)), [GPT-NeoX](https://github.com/EleutherAI/gpt-neox), and [Megatron-DeepSpeed](https://github.com/microsoft/Megatron-DeepSpeed/), with a flexible tokenization-agnostic interface.
- Support for commercial APIs including [OpenAI](https://openai.com), [goose.ai](https://goose.ai), and [TextSynth](https://textsynth.com/). - Support for commercial APIs including [OpenAI](https://openai.com), [goose.ai](https://goose.ai), and [TextSynth](https://textsynth.com/).
- Support for evaluation on adapters (e.g. LoRa) supported in [HuggingFace's PEFT library](https://github.com/huggingface/peft). - Support for evaluation on adapters (e.g. LoRa) supported in [HuggingFace's PEFT library](https://github.com/huggingface/peft).
......
...@@ -233,6 +233,7 @@ def create_all_tasks(): ...@@ -233,6 +233,7 @@ def create_all_tasks():
try: try:
resources_dir = importlib.resources.files("lm_eval.datasets") / "bigbench_resources" resources_dir = importlib.resources.files("lm_eval.datasets") / "bigbench_resources"
except: except:
import importlib_resources
resources_dir = importlib_resources.files("lm_eval.datasets") / "bigbench_resources" resources_dir = importlib_resources.files("lm_eval.datasets") / "bigbench_resources"
supported_tasks = [os.path.splitext(x)[0] for x in os.listdir(resources_dir)] supported_tasks = [os.path.splitext(x)[0] for x in os.listdir(resources_dir)]
res = {} res = {}
......
...@@ -25,6 +25,7 @@ setuptools.setup( ...@@ -25,6 +25,7 @@ setuptools.setup(
install_requires=[ install_requires=[
"datasets>=2.0.0", "datasets>=2.0.0",
"einops", "einops",
"importlib-resources",
"jsonlines", "jsonlines",
"numexpr", "numexpr",
"openai>=0.6.4", "openai>=0.6.4",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment