chong.zhang commited on
Commit
51dda1b
·
1 Parent(s): b5adbf2
Files changed (1) hide show
  1. app.py +5 -1
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()