diff --git a/requirements.txt b/requirements.txt index 50a5ca52..1d882737 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,7 +7,7 @@ evaluate>=0.3.0 fairscale faiss_gpu==1.7.2 jieba -mmengine>0.8.0 +mmengine>=0.8.2 nltk==3.8 numpy==1.23.4 openai diff --git a/run.py b/run.py index f671a3e8..479906f7 100644 --- a/run.py +++ b/run.py @@ -139,7 +139,7 @@ def main(): # initialize logger logger = get_logger(log_level='DEBUG' if args.debug else 'INFO') - cfg = Config.fromfile(args.config) + cfg = Config.fromfile(args.config, format_python_code=False) if args.work_dir is not None: cfg['work_dir'] = args.work_dir else: @@ -172,7 +172,7 @@ def main(): cfg.dump(output_config_path) # Config is intentally reloaded here to avoid initialized # types cannot be serialized - cfg = Config.fromfile(output_config_path) + cfg = Config.fromfile(output_config_path, format_python_code=False) # report to lark bot if specify --lark if not args.lark: