Update 2 files

- /export.py
- /trainer.cli.py

export.py

@@ -236,12 +236,11 @@ def ExportConfig(model, config, output_path):
 
 
 
-def ExportAll(model, tokenizer):
-    model.save_pretrained()
+def ExportAll():
     model = LoadModel('pytorch_model.bin')
     config = LoadConfig('config.json')
 
-    tokenizer.to_file('tokenizer.bin')
+    
 
     ExportModelFP32(model, config, 'model.fp32.bin')
     ExportModelINT8(model, config, 'model.int8.bin')

trainer.cli.py

@@ -56,4 +56,6 @@ if __name__ == '__main__':
     trainer.train(batches)
 
 
-    ExportAll(model, tokenizer)
+    model.save_pretrained()
+    tokenizer.to_file('tokenizer.bin')
+    ExportAll()