Skip to content

GitLab

Unverified Commit 221186c7 authored by Stella Biderman's avatar Stella Biderman Committed by GitHub
Browse files

Merge pull request #418 from zanussbaum/master 

fix: add seq2seq peft
parents 814940e8 1881deb6
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
lm_eval/models/huggingface.py
View file @ 221186c7
......@@ -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