mirror of https://github.com/hpcaitech/ColossalAI
fix torch 2.0 compatibility (#3346)
parent
8706a8c66c
commit
e78a1e949a
|
@ -8,4 +8,4 @@ click
|
||||||
fabric
|
fabric
|
||||||
contexttimer
|
contexttimer
|
||||||
ninja
|
ninja
|
||||||
torch>=1.11,<2.0
|
torch>=1.11
|
||||||
|
|
Loading…
Reference in New Issue