rahul7star commited on
Commit
2b876e9
·
verified ·
1 Parent(s): b3d298b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  # Copyright 2024-2025 The Alibaba Wan Team Authors. All rights reserved.
2
  # PyTorch 2.8 (temporary hack)
3
- import os
4
  os.system('pip install --upgrade --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu126 "torch<2.9" spaces')
5
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
6
  import spaces
 
1
  # Copyright 2024-2025 The Alibaba Wan Team Authors. All rights reserved.
2
  # PyTorch 2.8 (temporary hack)
3
+ import os,subprocess
4
  os.system('pip install --upgrade --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu126 "torch<2.9" spaces')
5
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
6
  import spaces