Spaces:
Sleeping
Sleeping
| import os | |
| def save_model_and_tokenizer(model, tokenizer, save_directory): | |
| """ | |
| Save model and tokenizer to the specified directory. | |
| Args: | |
| - model: The model to save. | |
| - tokenizer: The tokenizer to save. | |
| - save_directory: Directory where the model and tokenizer should be saved. | |
| """ | |
| try: | |
| # Ensure the save directory exists | |
| os.makedirs(save_directory, exist_ok=True) | |
| # Save model and tokenizer | |
| model.save_pretrained(save_directory, safe_serialization=True) | |
| tokenizer.save_pretrained(save_directory) | |
| print(f"Model and tokenizer saved locally at {save_directory}") | |
| except Exception as e: | |
| print(f"Error saving model and tokenizer: {str(e)}") | |
| raise | |