fix search space

Author: riapush

tests/assets/configs/multiclass.yaml

@@ -36,7 +36,7 @@
       learning_rate: [5.0e-5]
       seed: [0]
     - module_name: lora
-      model_config:
+      classification_model_config:
         - model_name: avsolatorio/GIST-small-Embedding-v0
       num_train_epochs: [1]
       batch_size: [8, 16]

tests/assets/configs/multilabel.yaml

@@ -32,7 +32,7 @@
       learning_rate: [5.0e-5]
       seed: [0]
     - module_name: lora
-      model_config:
+      classification_model_config:
         - model_name: avsolatorio/GIST-small-Embedding-v0
       num_train_epochs: [1]
       batch_size: [8]