Signed-off-by: Hanjun Cho <gkswns0531@gmail.com> Co-authored-by: wang.yuqi <yuqi.wang@daocloud.io>
torch.cuda.empty_cache
torch.accelerator.empty_cache