Spaces:
Runtime error
Runtime error
catch when no cuda device is available, fallback to CPU
Browse files- clip_for_ppts.py +1 -0
- requirements.txt +2 -2
clip_for_ppts.py
CHANGED
@@ -20,6 +20,7 @@ class ClipImage:
|
|
20 |
:param mode: 'image' or 'text' based on the type of input
|
21 |
:param device: device to run the model on
|
22 |
"""
|
|
|
23 |
# Path
|
24 |
directory = 'input_features'
|
25 |
path = os.path.join(path_to_save_image_features, directory)
|
|
|
20 |
:param mode: 'image' or 'text' based on the type of input
|
21 |
:param device: device to run the model on
|
22 |
"""
|
23 |
+
device = "cuda" if torch.cuda.is_available() else "cpu"
|
24 |
# Path
|
25 |
directory = 'input_features'
|
26 |
path = os.path.join(path_to_save_image_features, directory)
|
requirements.txt
CHANGED
@@ -5,9 +5,9 @@ sentence-transformers
|
|
5 |
pandas
|
6 |
langchain
|
7 |
gputil
|
8 |
-
git+https://github.com/openai/CLIP.git
|
9 |
torch
|
10 |
transformers
|
11 |
ftfy
|
12 |
regex
|
13 |
-
tqdm
|
|
|
|
5 |
pandas
|
6 |
langchain
|
7 |
gputil
|
|
|
8 |
torch
|
9 |
transformers
|
10 |
ftfy
|
11 |
regex
|
12 |
+
tqdm
|
13 |
+
git+https://github.com/openai/CLIP.git
|