pytorch / torchtitan

A native PyTorch Library for large model training
BSD 3-Clause "New" or "Revised" License
1.28k stars 115 forks source link

Add torchdata to requirements after release #351

Open gokulavasan opened 1 month ago

gokulavasan commented 1 month ago

Currently the user has to pip install the nightly explicitly. Instead once torchdata is released with StatefulDataLoader, add it to requirements.txt and pyproject.toml and remove the pip install from added in github workflow .yaml files and README Referenced here.