Skip to content

GitLab

Commit 1881deb6 authored by Zach Nussbaum's avatar Zach Nussbaum
Browse files

fix: add seq2seq peft

parent 83c2ff16
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
lm_eval/models/huggingface.py
View file @ 1881deb6
......@@ -467,6 +467,7 @@ class AutoSeq2SeqLM(HuggingFaceAutoLM):
"""
AUTO_MODEL_CLASS = transformers.AutoModelForSeq2SeqLM
AUTO_PEFT_CLASS = peft.PeftModel
@property
def max_length(self) -> int:
......
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