Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
requirements-rocm.txt 238 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
12
13
14
15
16
17 
tensorizer>=2.9.0

torch == 2.1.0
triton == 2.1.0
flash_attn == 2.0.4
xformers == 0.0.25
lmslim == 0.1.0