Spaces:
Runtime error
Runtime error
torch was incompatible when upgraded
Browse files- app.py +4 -0
- requirements.txt +3 -1
app.py
CHANGED
@@ -8,6 +8,10 @@ import logging
|
|
8 |
import spaces
|
9 |
from functools import lru_cache
|
10 |
|
|
|
|
|
|
|
|
|
11 |
# Set up logging
|
12 |
logging.basicConfig(level=logging.INFO)
|
13 |
logger = logging.getLogger(__name__)
|
|
|
8 |
import spaces
|
9 |
from functools import lru_cache
|
10 |
|
11 |
+
print(f"Is CUDA available: {torch.cuda.is_available()}")
|
12 |
+
# True
|
13 |
+
print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
|
14 |
+
|
15 |
# Set up logging
|
16 |
logging.basicConfig(level=logging.INFO)
|
17 |
logger = logging.getLogger(__name__)
|
requirements.txt
CHANGED
@@ -1,4 +1,6 @@
|
|
1 |
accelerate
|
2 |
transformers==4.42.4
|
3 |
SentencePiece
|
4 |
-
bitsandbytes
|
|
|
|
|
|
1 |
accelerate
|
2 |
transformers==4.42.4
|
3 |
SentencePiece
|
4 |
+
bitsandbytes
|
5 |
+
--extra-index-url https://download.pytorch.org/whl/cu113
|
6 |
+
torch
|