fix: add missing adaption map

custom_st.py

@@ -64,6 +64,9 @@ class Transformer(nn.Module):
             raise ValueError(
                 f"`lora_adaptations` must be a list and contain at least one element"
             )
+        self._adaptation_map = {
+            name: idx for idx, name in enumerate(self._lora_adaptations)
+        }
 
         if max_seq_length is not None and "model_max_length" not in tokenizer_args:
             tokenizer_args["model_max_length"] = max_seq_length