Shwu

AttributeError: Can't pickle local object 'add_hook_to_module.<locals>.new_forward'

1

Traceback (most recent call last): File "/home/shwu/LABS/FLAP/main.py", line 147, in main() File "/home/shwu/LABS/FLAP/main.py", line 141, in main torch.save(model, f'{args.save_model}/pruned_model.pt') File "/home/shwu/LABS/FLAP/venv/lib/python3.10/site-packages/torch/serialization.py", line 629, in save _save(obj, opened_zipfile, pickle_module, pickle_protocol, _disable_byteorder_record)...

Not able to install locally

8

### System Info ```shell 2024-04-22T09:19:51.209245Z INFO text_generation_launcher: Runtime environment: Target: x86_64-unknown-linux-gnu Cargo version: 1.75.0 Commit sha: N/A Docker label: N/A nvidia-smi: Mon Apr 22 09:19:50 2024 +-----------------------------------------------------------------------------+ | NVIDIA-SMI 525.105.17...