Files changed (3) hide show
  1. README.md +1 -1
  2. app.py +3 -0
  3. requirements.txt +2 -1
README.md CHANGED
@@ -4,7 +4,7 @@ emoji: 📊
4
  colorFrom: gray
5
  colorTo: yellow
6
  sdk: gradio
7
- sdk_version: 5.49.1
8
  app_file: app.py
9
  pinned: false
10
  license: apache-2.0
 
4
  colorFrom: gray
5
  colorTo: yellow
6
  sdk: gradio
7
+ sdk_version: 5.42.0
8
  app_file: app.py
9
  pinned: false
10
  license: apache-2.0
app.py CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  import spaces
2
 
3
  import argparse
 
1
+ import subprocess
2
+ subprocess.run('pip install flash-attn==2.7.0.post2 --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
3
+
4
  import spaces
5
 
6
  import argparse
requirements.txt CHANGED
@@ -1,4 +1,5 @@
 
1
  transformers==4.51.3
 
2
  pillow==10.3.0
3
  moviepy==1.0.3
4
- numpy<2
 
1
+ torch==2.4.0
2
  transformers==4.51.3
3
+ numpy==1.25.0
4
  pillow==10.3.0
5
  moviepy==1.0.3