quick fix for gpt export_model (PaddlePaddle#1001)

ruiwenfan · Sep 9, 2021 · b55782c · b55782c
1 parent 2d4dc85
commit b55782c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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