Currently pinned to datasets==1.11.0, torch==1.7.1, and a particular hugginface commit near 4.12.0. Updating to newer versions is currently out of scope for this work.