MAGNET training raises "No Module Error"

[fablos]

Description
When attempting to train MAGNet, the following error is encountered:

No module named 'torch.distributed.tensor'

Proposed Solution
The issue appears to be resolved by explicitly adding the following dependency specification for the transformers library in the env.yaml file:

dependencies:

• pip:
  • transformers==4.38.2