fix: not properly reference config (#12)

- fix: not properly reference config (a65808d3e03b5c65ad9fcc03a778ec3b86165caf)

custom_st.py

@@ -53,10 +53,10 @@ class Transformer(nn.Module):
         if config_args is None:
             config_args = {}
 
-        config = AutoConfig.from_pretrained(model_name_or_path, **config_args, cache_dir=cache_dir)
-        self.auto_model = AutoModel.from_pretrained(model_name_or_path, config=config, cache_dir=cache_dir, **model_args)
+        self.config = AutoConfig.from_pretrained(model_name_or_path, **config_args, cache_dir=cache_dir)
+        self.auto_model = AutoModel.from_pretrained(model_name_or_path, config=self.config, cache_dir=cache_dir, **model_args)
 
-        self._lora_adaptations = config.lora_adaptations
+        self._lora_adaptations = self.config.lora_adaptations
         if (
             not isinstance(self._lora_adaptations, list)
             or len(self._lora_adaptations) < 1