I consistently get the following error when running with torch v2.6.0, which is seemingly an allowed version in the pyproject.toml.
TypeError: DefaultSavePlanner.__init__() got an unexpected keyword argument 'enable_plan_caching'
It seems like that keyword is not available in torch v2.6.0.
https://github.com/pytorch/pytorch/blob/1eba9b3aa3c43f86f4a2c807ac8e12c4a7767340/torch/distributed/checkpoint/default_planner.py#L66
I consistently get the following error when running with torch v2.6.0, which is seemingly an allowed version in the
pyproject.toml.TypeError: DefaultSavePlanner.__init__() got an unexpected keyword argument 'enable_plan_caching'It seems like that keyword is not available in torch v2.6.0.
https://github.com/pytorch/pytorch/blob/1eba9b3aa3c43f86f4a2c807ac8e12c4a7767340/torch/distributed/checkpoint/default_planner.py#L66