Skip to content
This repository has been archived by the owner on Mar 8, 2024. It is now read-only.

Commit

Permalink
freeze torch<2.2.0 for FSDP compatibility (#27)
Browse files Browse the repository at this point in the history
  • Loading branch information
@Borda
Borda authored Feb 1, 2024
1 parent 70adaa3 commit 1c6516c
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions requirements.txt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3,3 +3,4 @@ anyio ==4.2.0
lightning >=2.1.0
six >=1.15.0
hetu-galvatron ==0.0.2
torch >=2.0.0, <2.2.0 # fix FSDP compatibility

0 comments on commit 1c6516c

Please sign in to comment.