Spaces:
Running
on
Zero
Running
on
Zero
chong.zhang
commited on
Commit
·
51dda1b
1
Parent(s):
b5adbf2
update
Browse files
app.py
CHANGED
@@ -26,6 +26,10 @@ import sys
|
|
26 |
|
27 |
os.system('nvidia-smi')
|
28 |
os.system('apt update -y && apt-get install -y apt-utils && apt install -y unzip')
|
|
|
|
|
|
|
|
|
29 |
print(torch.backends.cudnn.version())
|
30 |
|
31 |
ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
|
@@ -183,5 +187,5 @@ def main():
|
|
183 |
demo.launch()
|
184 |
|
185 |
if __name__ == '__main__':
|
186 |
-
subprocess.run(['bash', 'install.sh'], shell=True)
|
187 |
main()
|
|
|
26 |
|
27 |
os.system('nvidia-smi')
|
28 |
os.system('apt update -y && apt-get install -y apt-utils && apt install -y unzip')
|
29 |
+
os.system('pip install flash-attn --no-build-isolation')
|
30 |
+
os.system('git submodule update --init --recursive')
|
31 |
+
os.system('mkdir pretrained_models && cd pretrained_models && git clone https://huggingface.co/FunAudioLLM/InspireMusic-Base.git &&git clone https://huggingface.co/FunAudioLLM/InspireMusic-1.5B-Long.git &&git clone https://huggingface.co/FunAudioLLM/InspireMusic-1.5B.git &&git clone https://huggingface.co/FunAudioLLM/InspireMusic-1.5B-24kHz.git &&git clone https://huggingface.co/FunAudioLLM/InspireMusic-Base-24kHz.git && for i in InspireMusic-Base InspireMusic-Base-24kHz InspireMusic-1.5B InspireMusic-1.5B-24kHz InspireMusic-1.5B-Long; do sed -i -e "s/\.\.\/\.\.\///g" ${i}/inspiremusic.yaml; done')
|
32 |
+
|
33 |
print(torch.backends.cudnn.version())
|
34 |
|
35 |
ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
|
187 |
demo.launch()
|
188 |
|
189 |
if __name__ == '__main__':
|
190 |
+
# subprocess.run(['bash', 'install.sh'], shell=True)
|
191 |
main()
|