kadirnar commited on
Commit
7bf8cc1
1 Parent(s): e3aafd6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -28,6 +28,8 @@ def download_model(repo_id, model_name):
28
 
29
  import glob
30
 
 
 
31
  @spaces.GPU(duration=280)
32
  def run_inference(prompt_text):
33
  repo_id = "hpcai-tech/Open-Sora"
 
28
 
29
  import glob
30
 
31
+ subprocess.run(shlex.split("pip install flash-attn --no-build-isolation"))
32
+
33
  @spaces.GPU(duration=280)
34
  def run_inference(prompt_text):
35
  repo_id = "hpcai-tech/Open-Sora"