RuntimeError: Overflow when unpacking long

Environment info

• Machine: Google Cloud TPU VM version v2-alpha
• transformers: 4.18.0
• accelerate: 0.9.0.dev0 (same error happen with 0.8.0.dev0)

Script

I am training a GPT2 model using Pytorch run_clm_no_trainer.py.

Error

Below error happen when model is saving checkpoints. But seem that it only occurs at third or second checkpoint.

Traceback (most recent call last):
  File "/usr/local/lib/python3.8/dist-packages/torch_xla/distributed/xla_multiprocessing.py", line 329, in _mp_start_fn
    _start_fn(index, pf_cfg, fn, args)
  File "/usr/local/lib/python3.8/dist-packages/torch_xla/distributed/xla_multiprocessing.py", line 323, in _start_fn
    fn(gindex, *args)
  File "/usr/local/lib/python3.8/dist-packages/accelerate/utils/launch.py", line 55, in __call__
    self.launcher(*args)
  File "/home/nguyenhuuthuat09/gpt2/train_v1.py", line 553, in main
    accelerator.save_state(output_dir)      <---- this is line 564 in the original run_clm_no_trainer.py
  File "/usr/local/lib/python3.8/dist-packages/accelerate/accelerator.py", line 799, in save_state
    save_location = save_accelerator_state(
  File "/usr/local/lib/python3.8/dist-packages/accelerate/checkpointing.py", line 105, in save_accelerator_state
    states["xm_seed"] = torch.tensor(xm.get_rng_state())
RuntimeError: Overflow when unpacking long
Exception in device=TPU:0: Overflow when unpacking long

Enviroment variable

• export XRT_TPU_CONFIG="localservice;0;localhost:51011"
• I run accelerate config and use accelerate launch to run the code.
• After the error happen, I tried two below command but it doesn’t help. export XLA_USE_BF16=1 export XLA_TENSOR_ALLOCATOR_MAXSIZE=100000000

Releated issue

• I think this issue might be related to: https://github.com/huggingface/transformers/issues/10212
• I think the problem is on this line: https://github.com/huggingface/accelerate/blob/main/src/accelerate/checkpointing.py#L105
• I guess states["xm_seed"] = torch.tensor(xm.get_rng_state()) -> states["xm_seed"] = torch.tensor(xm.get_rng_state(), dtype=torch.float32) may help?

Thank you for great library!

[sgugger]

The seed is an int, not a float @nguyenhuuthuat09, you won’t be able to reload that RNG state if you save it as float.

The proper fix is to jsut remove torch.tensor here.

[sgugger]

Not sure why it’s wrapped inside a Tensor in the first place, @muellerzr ?