Update file export.py
Browse files
export.py
CHANGED
|
@@ -237,7 +237,7 @@ def ExportConfig(model, config, output_path):
|
|
| 237 |
|
| 238 |
|
| 239 |
def ExportAll(model, tokenizer):
|
| 240 |
-
model.
|
| 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 |
|