Open YanxiZSQ opened 8 months ago
I used post_pruning_processing.py to process the trained model. When executing model.prune_params(), an error occurred. The error is as follows:
Hi! Could you provide the full error, e.g., which line of the code triggered this error?
I run the code: python3 -m llmshearing.utils.post_pruning_processing prune_and_save_model model_file
I used post_pruning_processing.py to process the trained model. When executing model.prune_params(), an error occurred. The error is as follows:![image](https://github.com/princeton-nlp/LLM-Shearing/assets/152595968/b5324d65-4832-44ea-8232-d984fd795948)