11 lines
764 B
Plaintext
11 lines
764 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" and platform_system != "Darwin" # strict
|
|
bitsandbytes >=0.44.0,<0.44.2; sys_platform == 'linux' or sys_platform == 'win32'
|
|
bitsandbytes >=0.42.0,<0.43.0 ; sys_platform == 'darwin'
|