[diffusion] post_training: run weight update under torch.inference_mode()#31263

mickqian merged 6 commits into

sgl-project:mainsgl-project/sglang:mainfrom

kangwangamd:diffusion/weight-update-inference-modekangwangamd/sglang:diffusion/weight-update-inference-modeCopy head branch name to clipboard