fix: remove config args but pass config when initi auto model (#6)

- fix: remove config args but pass config when initi auto model (2da9381b5f52128f5e6284731aff667651282423)

custom_st.py

@@ -54,7 +54,7 @@ class Transformer(nn.Module):
             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_args, cache_dir=cache_dir, **model_args)
+        self.auto_model = AutoModel.from_pretrained(model_name_or_path, config=config, cache_dir=cache_dir, **model_args)
 
         if max_seq_length is not None and "model_max_length" not in tokenizer_args:
             tokenizer_args["model_max_length"] = max_seq_length