getting `_pickle.UnpicklingError: invalid load key, '<'.` while running awf

I've installed this repo on my server , and pulled model of awf, and running the project with all default settings using
```
python -m olmoearth_projects.main olmoearth_run olmoearth_run --config_path $PWD/olmoearth_run_data/awf/ --checkpoint_path $PWD/checkpoints/awf.ckpt --scratch_path project_data/awf/
```

it starts running the script but after few minutes it is giving error as

```
File "/opt/krishna/olmoearth_projects/.venv/lib/python3.12/site-packages/lightning/pytorch/strategies/strategy.py", line 368, in load_checkpoint
return self.checkpoint_io.load_checkpoint(checkpoint_path)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/krishna/olmoearth_projects/.venv/lib/python3.12/site-packages/lightning/fabric/plugins/io/torch_io.py", line 83, in load_checkpoint
return pl_load(path, map_location=map_location)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/krishna/olmoearth_projects/.venv/lib/python3.12/site-packages/lightning/fabric/utilities/cloud_io.py", line 60, in _load
return torch.load(
^^^^^^^^^^^
File "/opt/krishna/olmoearth_projects/.venv/lib/python3.12/site-packages/torch/serialization.py", line 1549, in load
return _legacy_load(
^^^^^^^^^^^^^
File "/opt/krishna/olmoearth_projects/.venv/lib/python3.12/site-packages/torch/serialization.py", line 1797, in _legacy_load
magic_number = pickle_module.load(f, **pickle_load_args)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
_pickle.UnpicklingError: invalid load key, '<'.
```