Signed-off-by: esmeetu <jasonailu87@gmail.com> Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
torch.cuda.empty_cache
torch.accelerator.empty_cache