Skip to content

GitLab

Commit db1f55ff authored by haileyschoelkopf's avatar haileyschoelkopf
Browse files

first step: add HF+accelerate implementations

parent 25780307
Expand all Hide whitespace changes
Inline Side-by-side
Showing with 639 additions and 0 deletions
lm_eval/models/__init__.py
View file @ db1f55ff
from . import gpt2 from . import gpt2
from . import gpt3 from . import gpt3
from . import huggingface
from . import textsynth from . import textsynth
from . import dummy from . import dummy
MODEL_REGISTRY = { MODEL_REGISTRY = {
"hf": gpt2.HFLM, "hf": gpt2.HFLM,
"hf-causal": huggingface.AutoCausalLM,
"hf-seq2seq": huggingface.AutoSeq2SeqLM,
"gpt2": gpt2.GPT2LM, "gpt2": gpt2.GPT2LM,
"gpt3": gpt3.GPT3LM, "gpt3": gpt3.GPT3LM,
"textsynth": textsynth.TextSynthLM, "textsynth": textsynth.TextSynthLM,
......
lm_eval/models/huggingface.py 0 → 100644
View file @ db1f55ff
This diff is collapsed.
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