Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
requirements-rocm.txt 221 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
Kevin H. Luu's avatar
[ci/build] Fix AMD CI dependencies (#11087)  
Kevin H. Luu committed
8 
datasets
Alexei-V-Ivanov-AMD's avatar
[Build/CI] Enabling AMD Entrypoints Test (#4834)  
Alexei-V-Ivanov-AMD committed
9 
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
10 
peft
Alexei-V-Ivanov-AMD's avatar
[Build/CI] Enabling AMD Entrypoints Test (#4834)  
Alexei-V-Ivanov-AMD committed
11 
pytest-asyncio
Kevin H. Luu's avatar
[ci/build] Fix AMD CI dependencies (#11087)  
Kevin H. Luu committed
12 
tensorizer>=2.9.0
Kevin H. Luu's avatar
[ci] Add AWS creds for AMD (#13572)  
Kevin H. Luu committed
13 
runai-model-streamer==0.11.0
TJian's avatar
[ROCm] Faster Custom Paged Attention kernels (#12348)  
TJian committed
14 
runai-model-streamer-s3==0.11.0