Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
requirements-rocm.txt 261 Bytes
Newer Older
Woosuk Kwon's avatar
[Misc] Define common requirements (#3841)  
Woosuk Kwon committed
1
2
3
4 
# Common dependencies
-r requirements-common.txt

# Dependencies for AMD GPUs
Matt Wong's avatar
[Bugfix][CI/Build][Hardware][AMD] Fix AMD tests, add HF cache, update CK FA,...  
Matt Wong committed
5
6
7 
awscli
boto3
botocore
Alexei-V-Ivanov-AMD's avatar
[Build/CI] Enabling AMD Entrypoints Test (#4834)  
Alexei-V-Ivanov-AMD committed
8 
ray >= 2.10.0
Matt Wong's avatar
[Bugfix][CI/Build][Hardware][AMD] Fix AMD tests, add HF cache, update CK FA,...  
Matt Wong committed
9 
peft
Alexei-V-Ivanov-AMD's avatar
[Build/CI] Enabling AMD Entrypoints Test (#4834)  
Alexei-V-Ivanov-AMD committed
10 
pytest-asyncio
zhuwenwen's avatar
update version and requirements-rocm.txt  
zhuwenwen committed
11 
tensorizer>=2.9.0
zhuwenwen's avatar
update benchmarks  
zhuwenwen committed
12 
setuptools_scm>=8
zhuwenwen's avatar
update version and requirements-rocm.txt  
zhuwenwen committed
13
zhuwenwen's avatar
update torch version and support lm_head nn layout  
zhuwenwen committed
14 
torch == 2.3.0
zhuwenwen's avatar
update version and requirements-rocm.txt  
zhuwenwen committed
15 
triton == 2.1.0
zhuwenwen's avatar
update torch version and support lm_head nn layout  
zhuwenwen committed
16 
flash_attn == 2.6.1
zhuwenwen's avatar
remove xformers deps  
zhuwenwen committed
17 
lmslim == 0.1.2  # future version 0.2.0