RuntimeError: mmap can only be used with files saved with `torch.save(./ckpts/seedvr2_ema_3b.pth, _use_new_zipfile_serialization=True)

[rank0]: Traceback (most recent call last):
[rank0]: File "/inspire/ssd/project/video-generation/public/zxu/SeedVR/projects/inference_seedvr2_3b.py", line 336, in
[rank0]: runner = configure_runner(args.sp_size)
[rank0]: File "/inspire/ssd/project/video-generation/public/zxu/SeedVR/projects/inference_seedvr2_3b.py", line 77, in configure_runner
[rank0]: runner.configure_dit_model(device="cuda", checkpoint='./ckpts/seedvr2_ema_3b.pth')
[rank0]: File "/inspire/ssd/project/video-generation/public/zxu/SeedVR/common/decorators.py", line 39, in log_on_entry_wrapper
[rank0]: return func(*args, **kwargs)
[rank0]: File "/inspire/ssd/project/video-generation/public/zxu/SeedVR/common/decorators.py", line 142, in wrapped
[rank0]: result = func(*args, **kwargs)
[rank0]: File "/inspire/ssd/project/video-generation/public/zxu/SeedVR/projects/video_diffusion_sr/infer.py", line 89, in configure_dit_model
[rank0]: state = torch.load(checkpoint, map_location="cpu", mmap=True)
[rank0]: File "/root/miniconda3/envs/seedvr_py310/lib/python3.10/site-packages/torch/serialization.py", line 1032, in load
[rank0]: raise RuntimeError("mmap can only be used with files saved with "
[rank0]: RuntimeError: mmap can only be used with files saved with `torch.save(./ckpts/seedvr2_ema_3b.pth, _use_new_zipfile_serialization=True), please torch.save your checkpoint with this option in order to use mmap.