PyRIT
PyRIT copied to clipboard
BUG GCG runs out of memory even on huge machines
Describe the bug
We're getting out of memory issues even on massive machines that have enough memory for several instances of the target language model. We need to investigate why and remediate the problem.
Steps/Code to Reproduce
Run GCG with 1000 steps. Note that this requires someone with experience with pytorch and running LLMs on compute.