diff --git a/README.md b/README.md
index 79406917..d6bb0b4a 100644
--- a/README.md
+++ b/README.md
@@ -22,7 +22,7 @@
 
 ## update
 - <strong>2023-05-19</strong>
-  - 0.1.6 release
+  - 0.1.7 release
   - fix 0.1.5 rl bugs
 - <strong>2023-05-10</strong>
   - 0.1.5 release
diff --git a/data_helper/data_helper.py b/data_helper/data_helper.py
index 5c3ef459..3175e3bc 100644
--- a/data_helper/data_helper.py
+++ b/data_helper/data_helper.py
@@ -88,7 +88,7 @@ def on_get_corpus(self, files: typing.List[str], mode: str):
 
 class DataHelper(DataPreprocessHelper):
     def __init__(self,model_args: ModelArguments,
-                training_args: TrainingArguments,
+                training_args: typing.Optional[TrainingArguments],
                 data_args: DataArguments,**kwargs):
         super(DataHelper, self).__init__()
 
diff --git a/nlp/models/lora/v1/configuration.py b/nlp/models/lora/v1/configuration.py
index b2632bc2..cabee028 100644
--- a/nlp/models/lora/v1/configuration.py
+++ b/nlp/models/lora/v1/configuration.py
@@ -74,6 +74,8 @@ def from_pretrained(cls, pretrained_model_name_or_path, **kwargs):
 
         loaded_attributes = cls.from_json_file(config_file)
 
+        kwargs.pop("subfolder", None)
+
         config = cls(**kwargs)
 
         for key, value in loaded_attributes.items():
diff --git a/nlp/models/lora/v2/configuration.py b/nlp/models/lora/v2/configuration.py
index f22efdb6..8a8fe478 100644
--- a/nlp/models/lora/v2/configuration.py
+++ b/nlp/models/lora/v2/configuration.py
@@ -77,6 +77,8 @@ def from_pretrained(cls, pretrained_model_name_or_path, **kwargs):
 
         loaded_attributes = cls.from_json_file(config_file)
 
+        kwargs.pop("subfolder",None)
+
         config = cls(**kwargs)
 
         for key, value in loaded_attributes.items():
diff --git a/nlp/models/prompt/configuration.py b/nlp/models/prompt/configuration.py
index 55e6739a..e90ac879 100644
--- a/nlp/models/prompt/configuration.py
+++ b/nlp/models/prompt/configuration.py
@@ -86,6 +86,8 @@ def from_pretrained(cls, pretrained_model_name_or_path, **kwargs):
 
         loaded_attributes = cls.from_json_file(config_file)
 
+        kwargs.pop("subfolder", None)
+
         config = cls(**kwargs)
 
         for key, value in loaded_attributes.items():
diff --git a/setup.py b/setup.py
index e1f0ed08..b05492ec 100644
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,7 @@
 ignore = ['test','tests']
 setup(
     name='deep_training',
-    version='0.1.6',
+    version='0.1.7',
     description='an easy training architecture',
     long_description='torch_training: https://github.com/ssbuild/deep_training.git',
     license='Apache License 2.0',