Jiax comments

Repositories
Issues
Comments

Results 2 comments of


                                            Jiax

Suggestions for PMET Method on Llama

Thanks for your reply 😊

The GPU memory will keep increase

Hi, I came through the same question. And this may be caused by the KnowledgeNeurons class not releasing its memory in time. You can try `del` and ` torch.cuda.empty_cache()`, like:...