Open six-m opened 1 year ago
The buffer named 'total_ops' and 'total_params' maybe remain inside the model after calling profile function when the submodule is not supported. And the checkpoints exported contains the buffers which affect loading.
This PR fix this.
The buffer named 'total_ops' and 'total_params' maybe remain inside the model after calling profile function when the submodule is not supported. And the checkpoints exported contains the buffers which affect loading.
This PR fix this.