Skip to content

Commit

Permalink
permit torch nightly
Browse files Browse the repository at this point in the history
  • Loading branch information
dakinggg committed Apr 20, 2024
1 parent 3426415 commit b030b5e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@
'accelerate>=0.25,<0.26', # for HF inference `device_map`
'transformers>=4.40,<4.41',
'mosaicml-streaming>=0.7.5,<0.8',
'torch>=2.2.1,<2.3',
'torch>=2.2.1,<2.3.1', # TODO: tighten before release
'datasets>=2.16,<2.17',
'fsspec==2023.6.0', # newer version results in a bug in datasets that duplicates data
'sentencepiece==0.1.97',
Expand Down

0 comments on commit b030b5e

Please sign in to comment.