Torch.multinomial with generator is not determinstic under certain conditions

2

This seems to be related to this post so make sure you are enabling deterministic algorithms and are using a new PyTorch version.