Update README.md
Browse files
README.md
CHANGED
|
@@ -61,7 +61,7 @@ def train_model(model, tokenizer, train_data, output_dir, epochs=3, batch_size=1
|
|
| 61 |
|
| 62 |
print(f"Epoch {epoch + 1} Loss: {epoch_loss / len(train_loader)}")
|
| 63 |
|
| 64 |
-
model.save_pretrained(output_dir)
|
| 65 |
tokenizer.save_pretrained(output_dir)
|
| 66 |
```
|
| 67 |
|
|
|
|
| 61 |
|
| 62 |
print(f"Epoch {epoch + 1} Loss: {epoch_loss / len(train_loader)}")
|
| 63 |
|
| 64 |
+
model.save_pretrained(output_dir, safe_serialization = False)
|
| 65 |
tokenizer.save_pretrained(output_dir)
|
| 66 |
```
|
| 67 |
|