Spaces:
Build error
Build error
add dependencies
Browse files- app.py +2 -0
- requirements.txt +1 -0
app.py
CHANGED
|
@@ -1,3 +1,5 @@
|
|
|
|
|
|
|
|
| 1 |
import spaces
|
| 2 |
import torch
|
| 3 |
import torch.nn.functional as F
|
|
|
|
| 1 |
+
# main.py
|
| 2 |
+
|
| 3 |
import spaces
|
| 4 |
import torch
|
| 5 |
import torch.nn.functional as F
|
requirements.txt
CHANGED
|
@@ -2,3 +2,4 @@ torch==2.2.0
|
|
| 2 |
transformers
|
| 3 |
flash-attn==2.2.0
|
| 4 |
einops
|
|
|
|
|
|
| 2 |
transformers
|
| 3 |
flash-attn==2.2.0
|
| 4 |
einops
|
| 5 |
+
sentence-transformers==2.7.0
|