search

pytorch / xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)
https://pytorch.org/xla
Other
2.49k stars 480 forks source link

Set weights_only to False for xser.load #8259

Closed JackCaoG closed 1 month ago

JackCaoG commented 1 month ago

should fix https://github.com/pytorch/xla/issues/7799

After https://github.com/pytorch/pytorch/pull/137602 the weights_only will default to True but it will fail this xla test.