From b55782c6c5e795153ae9252cb197c9ba6e32385e Mon Sep 17 00:00:00 2001 From: Zhong Hui Date: Thu, 9 Sep 2021 16:50:46 +0800 Subject: [PATCH] quick fix for gpt export_model (#1001) --- examples/language_model/gpt/export_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/language_model/gpt/export_model.py b/examples/language_model/gpt/export_model.py index 0278537c3d4e..0def3a231dce 100644 --- a/examples/language_model/gpt/export_model.py +++ b/examples/language_model/gpt/export_model.py @@ -62,7 +62,7 @@ def main(): model = model_class.from_pretrained( args.model_path, max_predict_len=32, - eol_token_id=self.tokenizer.eol_token_id) + eol_token_id=tokenizer.eol_token_id) # Switch to eval model model.eval() # Convert to static graph with specific input description