multimodalart HF Staff commited on
Commit
c61da4b
·
verified ·
1 Parent(s): b68f017

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -117,9 +117,7 @@ def torch_gc():
117
  torch.cuda.empty_cache()
118
  torch.cuda.ipc_collect()
119
 
120
- #subprocess.run('pip install flash-attn==2.7.4.post1 --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
121
-
122
- @spaces.GPU(duration=500)
123
  def generate_video(
124
  mode,
125
  prompt,
 
117
  torch.cuda.empty_cache()
118
  torch.cuda.ipc_collect()
119
 
120
+ @spaces.GPU(duration=900)
 
 
121
  def generate_video(
122
  mode,
123
  prompt,