model.prune_params() NotImplementedError: Could not run 'aten::nonzero'

[YanxiZSQ]

I used post_pruning_processing.py to process the trained model. When executing model.prune_params(), an error occurred. The error is as follows:

[xiamengzhou]

Hi! Could you provide the full error, e.g., which line of the code triggered this error?

[YanxiZSQ]

[YanxiZSQ]

I run the code:
python3 -m llmshearing.utils.post_pruning_processing prune_and_save_model model_file