lightning/requirements/fabric/strategies.txt

10 lines
625 B
Plaintext

# Despite its misleading name, this file contains dependencies for testing that only get installed in the GPU CI
# NOTE: the upper bound for the package version is only set for CI stability, and it is dropped while installing this package
# in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
# note: is a bug around 0.10 with `MPS_Accelerator must implement all abstract methods`
# shall be resolved by https://github.com/microsoft/DeepSpeed/issues/4372
deepspeed >=0.8.2, <=0.9.3; platform_system != "Windows" # strict
bitsandbytes ==0.41.0 # strict