mlx.nn.Module.save_weights

mlx.nn.Module.save_weights#

Module.save_weights(file: str)#

Save the model’s weights to a file. The saving method is determined by the file extension: - .npz will use mx.savez() - .safetensors will use mx.save_safetensors()