diff --git a/tests/test_config.py b/tests/test_config.py index 0590b4fd04..70bcc56101 100644 --- a/tests/test_config.py +++ b/tests/test_config.py @@ -62,6 +62,7 @@ TaskType, TinyLoraConfig, TrainableTokensConfig, + UniLoraConfig, VBLoRAConfig, VeraConfig, WaveFTConfig, @@ -118,6 +119,7 @@ class TestingCommitHashError(Exception): (OFTConfig, {}), (PveraConfig, {}), (WaveFTConfig, {}), + (UniLoraConfig, {}), )