Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
requirements-rocm.txt 316 Bytes
Newer Older
Woosuk Kwon's avatar
[Misc] Define common requirements (#3841)  
Woosuk Kwon committed
1
2
3 
# Common dependencies
-r requirements-common.txt
Nishidha's avatar
Moved numba from common requirements to cuda/rocm specific requirements (#14199)  
Nishidha committed
4
5 
numba == 0.60.0 # v0.61 doesn't support Python 3.9. Required for N-gram speculative decoding
Woosuk Kwon's avatar
[Misc] Define common requirements (#3841)  
Woosuk Kwon committed
6 
# 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
7
8
9 
awscli
boto3
botocore
Kevin H. Luu's avatar
[ci/build] Fix AMD CI dependencies (#11087)  
Kevin H. Luu committed
10 
datasets
Alexei-V-Ivanov-AMD's avatar
[Build/CI] Enabling AMD Entrypoints Test (#4834)  
Alexei-V-Ivanov-AMD committed
11 
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
12 
peft
Alexei-V-Ivanov-AMD's avatar
[Build/CI] Enabling AMD Entrypoints Test (#4834)  
Alexei-V-Ivanov-AMD committed
13 
pytest-asyncio
Kevin H. Luu's avatar
[ci/build] Fix AMD CI dependencies (#11087)  
Kevin H. Luu committed
14 
tensorizer>=2.9.0
Kevin H. Luu's avatar
[ci] Add AWS creds for AMD (#13572)  
Kevin H. Luu committed
15 
runai-model-streamer==0.11.0
Nishidha's avatar
Moved numba from common requirements to cuda/rocm specific requirements (#14199)  
Nishidha committed
16 
runai-model-streamer-s3==0.11.0