Spaces:
Paused
Paused
MaziyarPanahi
commited on
Commit
•
a533ef3
1
Parent(s):
02558d9
Update app.py (#15)
Browse files- Update app.py (f632f3faa08ba4ebadeb9de72e323beb1f565329)
app.py
CHANGED
@@ -6,6 +6,8 @@ import torch
|
|
6 |
from PIL import Image
|
7 |
import subprocess
|
8 |
from datetime import datetime
|
|
|
|
|
9 |
|
10 |
# subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
|
11 |
|
|
|
6 |
from PIL import Image
|
7 |
import subprocess
|
8 |
from datetime import datetime
|
9 |
+
import numpy as np
|
10 |
+
|
11 |
|
12 |
# subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
|
13 |
|