SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

TST PEFT integration tests with pipeline generate (#39086)

Browse Source 
Some PEFT integration tests involving text generation pipelines were
failing since #38129 because the base model is too small to generate
longer sequences. Setting max_new_tokens fixes this.
This commit is contained in:
Benjamin Bossan
2025-06-27 15:58:10 +02:00
committed by GitHub
parent 993665a5ff
commit 0c35280e58
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/peft_integration/test_peft_integration.py
View File

@@ -531,7 +531,7 @@ class PeftIntegrationTester(unittest.TestCase, PeftTesterMixin):
peft_params = list(peft_pipe.model.parameters()) peft_params = list(peft_pipe.model.parameters())
base_params = list(base_pipe.model.parameters()) base_params = list(base_pipe.model.parameters())
self.assertNotEqual(len(peft_params), len(base_params)) # Assert we actually loaded the adapter too self.assertNotEqual(len(peft_params), len(base_params)) # Assert we actually loaded the adapter too
_ = peft_pipe("Hello") _ = peft_pipe("Hello", max_new_tokens=20)
def test_peft_add_adapter_with_state_dict(self): def test_peft_add_adapter_with_state_dict(self):
""" """
@@ -858,4 +858,4 @@ class PeftIntegrationTester(unittest.TestCase, PeftTesterMixin):
) )
# Generate text to verify pipeline works # Generate text to verify pipeline works
_ = lora_generator(text) _ = lora_generator(text, max_new_tokens=20)