LynxDemo / requirements.txt
Allen Park
add accelerate pip package and add device_map='auto'
7d126a2
raw
history blame
53 Bytes
torch==2.4.0
transformers==4.43.3
accelerate==0.33.0