Skip to content

GitLab

"test/vscode:/vscode.git/clone" did not exist on "027e46ee82cdb8a02313985361333c78f281b887"
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): ...@@ -467,6 +467,7 @@ class AutoSeq2SeqLM(HuggingFaceAutoLM):
""" """
AUTO_MODEL_CLASS = transformers.AutoModelForSeq2SeqLM AUTO_MODEL_CLASS = transformers.AutoModelForSeq2SeqLM
AUTO_PEFT_CLASS = peft.PeftModel
@property @property
def max_length(self) -> int: 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