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

add and fix tests

Browse Source
This commit is contained in:
younesbelkada
2023-04-07 10:48:22 +00:00
parent c22a57420c
commit f35b20a845
5 changed files with 53 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_encoder_decoder_models.py
+4
View File
@@ -86,3 +86,7 @@ class PeftEncoderDecoderModelTester(unittest.TestCase, PeftCommonTester):
@parameterized.expand(PeftTestConfigManager.get_grid_parameters(FULL_GRID, filter_params_func=skip_non_lora_or_pt))
def test_generate(self, test_name, model_id, config_cls, config_kwargs):
self._test_generate(model_id, config_cls, config_kwargs)
@parameterized.expand(PeftTestConfigManager.get_grid_parameters(FULL_GRID))
def test_generate_half_prec(self, test_name, model_id, config_cls, config_kwargs):
self._test_generate_half_prec(model_id, config_cls, config_kwargs)