Spaces:
Build error
Build error
File size: 607 Bytes
31a7207 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
#!/bin/bash -e
test -f /etc/shinit_v2 && source /etc/shinit_v2
cd WhisperFusion
if [ "$1" != "mistral" ]; then
exec python3 main.py --phi \
--whisper_tensorrt_path /root/whisper_small_en \
--phi_tensorrt_path /root/dolphin-2_6-phi-2 \
--phi_tokenizer_path /root/dolphin-2_6-phi-2/tokenizer
else
exec python3 main.py --mistral \
--whisper_tensorrt_path /root/models/whisper_small_en \
--mistral_tensorrt_path /root/models/mistral \
--mistral_tokenizer_path teknium/OpenHermes-2.5-Mistral-7B
fi
|