Merge pull request #551 from FlagAI-Open/fix_issue

fixed yaml setup issue
FlagAI-Open · Oct 19, 2023 · 13e7191 · 13e7191
2 parents 0178066 + 551ac87
commit 13e7191
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 7 deletions.
diff --git a/examples/Aquila/Aquila-chat/aquila_chat.py b/examples/Aquila/Aquila-chat/aquila_chat.py
@@ -45,7 +45,7 @@
 if env_args.yaml_config:
     import yaml
     file_data = open(env_args.yaml_config, 'r', encoding="utf-8").read()
-    data = yaml.load_all(file_data)
+    data = yaml.load_all(file_data, Loader=yaml.SafeLoader)
     delattr(env_args, 'yaml_config')
     arg_dict = env_args.__dict__
     for subdata in data:

diff --git a/examples/Aquila/Aquila-code/aquila_code_pretrain.py b/examples/Aquila/Aquila-code/aquila_code_pretrain.py
@@ -45,7 +45,7 @@
 if env_args.yaml_config:
     import yaml
     file_data = open(env_args.yaml_config, 'r', encoding="utf-8").read()
-    data = yaml.load_all(file_data)
+    data = yaml.load_all(file_data, Loader=yaml.SafeLoader)
     delattr(env_args, 'yaml_config')
     arg_dict = env_args.__dict__
     for subdata in data:

diff --git a/examples/Aquila/Aquila-pretrain/aquila_pretrain.py b/examples/Aquila/Aquila-pretrain/aquila_pretrain.py
@@ -44,7 +44,7 @@
 if env_args.yaml_config:
     import yaml
     file_data = open(env_args.yaml_config, 'r', encoding="utf-8").read()
-    data = yaml.load_all(file_data)
+    data = yaml.load_all(file_data, Loader=yaml.SafeLoader)
     delattr(env_args, 'yaml_config')
     arg_dict = env_args.__dict__
     for subdata in data:

diff --git a/examples/Aquila/Aquila-sql/aquila_sql_flash.py b/examples/Aquila/Aquila-sql/aquila_sql_flash.py
@@ -50,7 +50,7 @@
 if env_args.yaml_config:
     import yaml
     file_data = open(env_args.yaml_config, 'r', encoding="utf-8").read()
-    data = yaml.load_all(file_data)
+    data = yaml.load_all(file_data, Loader=yaml.SafeLoader)
     delattr(env_args, 'yaml_config')
     arg_dict = env_args.__dict__
     for subdata in data:

diff --git a/examples/Aquila/aquila_pretrain.py b/examples/Aquila/aquila_pretrain.py
@@ -45,7 +45,7 @@
 if env_args.yaml_config:
     import yaml
     file_data = open(env_args.yaml_config, 'r', encoding="utf-8").read()
-    data = yaml.load_all(file_data)
+    data = yaml.load_all(file_data, Loader=yaml.SafeLoader)
     delattr(env_args, 'yaml_config')
     arg_dict = env_args.__dict__
     for subdata in data:

diff --git a/setup.py b/setup.py
@@ -27,7 +27,7 @@
         'transformers>=4.31.0',
         'datasets>=2.0.0',
         'setuptools>=66.0.0',
-        'protobuf==3.19.6',
+        'protobuf>=3.19.6',
         'ftfy',
         'Pillow>=9.3.0',
         'einops>=0.3.0',
@@ -38,7 +38,7 @@
         'sacrebleu>=2.3.1',
         'jsonlines',
         'accelerate',
-        'PyYAML==5.4.1',
+        'PyYAML>=5.4.1',
         'safetensors',
         'timm',
     ]