Signed-off-by: rahul-sarvam <140298821+rahul-sarvam@users.noreply.github.com>
torch.cuda.empty_cache
torch.accelerator.empty_cache