Update tests/model_zoo/test_gpt.py

Co-authored-by: Zhong Hui <zhonghui.net@gmail.com>
PaddlePaddle · Jan 31, 2023 · 6862118 · 6862118
1 parent 6aba70e
commit 6862118
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/model_zoo/test_gpt.py b/tests/model_zoo/test_gpt.py
@@ -40,19 +40,19 @@ def test_pretrain(self):
             do_train()
 
         # 2. export model
-        config = {
+        export_config = {
             "model_type": config["model_type"],
             "model_path": config["output_dir"],
             "output_path": os.path.join(config["output_dir"], "export_model"),
         }
-        with argv_context_guard(config):
+        with argv_context_guard(export_config):
             from export_model import main
 
             main()
 
         # 3. infer model
-        config = {"model_type": config["model_type"], "model_path": config["output_path"], "select_device": device}
-        with argv_context_guard(config):
+        infer_config = {"model_type": export_config["model_type"], "model_path": export_config["output_path"], "select_device": device}
+        with argv_context_guard(infer_config):
             from deploy.python.inference import main
 
             main()