flpelerin commited on
Commit
4f246b8
·
1 Parent(s): 44afb45

Update file export.py

Browse files
Files changed (1) hide show
  1. export.py +1 -1
export.py CHANGED
@@ -237,7 +237,7 @@ def ExportConfig(model, config, output_path):
237
 
238
 
239
  def ExportAll(model, tokenizer):
240
- model.save()
241
  model = LoadModel('pytorch_model.bin')
242
  config = LoadConfig('config.json')
243
 
 
237
 
238
 
239
  def ExportAll(model, tokenizer):
240
+ model.save_pretrained()
241
  model = LoadModel('pytorch_model.bin')
242
  config = LoadConfig('config.json')
243