wassname/peft
1
0
Fork 0
mirror of https://github.com/wassname/peft.git synced 2026-09-09 11:28:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix test

Browse Source
This commit is contained in:
younesbelkada
2023-04-05 17:17:01 +00:00
parent d936aa9349
commit 37e1f9ba34
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_encoder_decoder_models.py
+2 -2
View File
@@ -22,7 +22,7 @@ from .testing_common import PeftCommonTester, PeftTestConfigManager
PEFT_ENCODER_DECODER_MODELS_TO_TEST = [
"hf-internal-testing/tiny-random-T5ForConditionalGeneration",
"ybelkada/tiny-random-T5ForConditionalGeneration-calibrated",
"hf-internal-testing/tiny-random-BartForConditionalGeneration",
]
@@ -74,7 +74,7 @@ class PeftEncoderDecoderModelTester(unittest.TestCase, PeftCommonTester):
PeftTestConfigManager.get_grid_parameters(
{
"model_ids": PEFT_ENCODER_DECODER_MODELS_TO_TEST,
"lora_kwargs": {"init_lora_weights": [False], "merge_weights": [True, False]},
"lora_kwargs": {"init_lora_weights": [False], "merge_weights": [False, True]},
"task_type": "SEQ_2_SEQ_LM",
},
)