DRA-Ctrl / app.sh
caohy666's picture
<feat> support changing models.
931adb9
raw
history blame
140 Bytes
#!/bin/bash
pip install torch==2.5.1
pip install git+https://github.com/Dao-AILab/flash-attention.git@v2.7.3#subdirectory=csrc
python app.py