Commit
•
9cdab82
1
Parent(s):
05aabe7
Update app.py
Browse files
app.py
CHANGED
|
@@ -22,9 +22,9 @@ def readb64(b64):
|
|
| 22 |
return img
|
| 23 |
|
| 24 |
|
| 25 |
-
|
| 26 |
-
import subprocess
|
| 27 |
-
subprocess.run('pip3 install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
|
| 28 |
|
| 29 |
model_id = "vikhyatk/moondream2"
|
| 30 |
revision = "2024-04-02"
|
|
|
|
| 22 |
return img
|
| 23 |
|
| 24 |
|
| 25 |
+
# not sure why
|
| 26 |
+
#import subprocess
|
| 27 |
+
#subprocess.run('pip3 install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
|
| 28 |
|
| 29 |
model_id = "vikhyatk/moondream2"
|
| 30 |
revision = "2024-04-02"
|