0.1.7

Signed-off-by: ssbuild <[email protected]>

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

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__()
 

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():

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():

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():

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',