Spaces:
Running on Zero
Running on Zero
myOvis
#3
by fridayfairy - opened
- README.md +1 -1
- app.py +3 -0
- 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.
|
| 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
|
|
|