diff --git a/mamba_ssm/models/mixer_seq_simple.py b/mamba_ssm/models/mixer_seq_simple.py index 2f1d97fd8..cd224738d 100644 --- a/mamba_ssm/models/mixer_seq_simple.py +++ b/mamba_ssm/models/mixer_seq_simple.py @@ -252,8 +252,7 @@ def save_pretrained(self, save_directory): Save the model and its configuration file to a directory. """ # Ensure save_directory exists - if not os.path.exists(save_directory): - os.makedirs(save_directory) + os.makedirs(save_directory, exist_ok=True) # Save the model's state_dict model_path = os.path.join(save_directory, 'pytorch_model.bin')