2024-01-11 07:50:43 +00:00
|
|
|
# Despite its misleading name, this file contains dependencies for testing that only get installed in the GPU CI
|
|
|
|
|
|
2022-09-01 22:13:12 +00:00
|
|
|
# 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
|
2023-10-03 19:05:40 +00:00
|
|
|
|
|
|
|
|
# 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
|
2024-04-27 05:24:07 +00:00
|
|
|
deepspeed >=0.8.2, <=0.9.3; platform_system != "Windows" and platform_system != "Darwin" # strict
|
2024-10-02 16:18:43 +00:00
|
|
|
bitsandbytes >=0.44.0,<0.44.2; sys_platform == 'linux' or sys_platform == 'win32'
|
|
|
|
|
bitsandbytes >=0.42.0,<0.43.0 ; sys_platform == 'darwin'
|
